MMSegmentation语义分割工具箱

联合创作 · 2023-09-25 22:46

MMSegmentation 是一个基于 PyTorch 的语义分割开源工具箱。它是 OpenMMLab 项目的一部分,目前支持 PyTorch 1.5 以上的版本。

主要特性

  • 统一的基准平台

    我们将各种各样的语义分割算法集成到了一个统一的工具箱,进行基准测试。

  • 模块化设计

    MMSegmentation 将分割框架解耦成不同的模块组件,通过组合不同的模块组件,用户可以便捷地构建自定义的分割模型。

  • 丰富的即插即用的算法和模型

    MMSegmentation 支持了众多主流的和最新的检测算法,例如 PSPNet,DeepLabV3,PSANet,DeepLabV3+ 等.

  • 速度快

    训练速度比其他语义分割代码库更快或者相当。

基准测试和模型库

测试结果和模型可以在模型库中找到。

已支持的骨干网络:

已支持的算法:

已支持的数据集:

浏览 8
点赞
评论
收藏
分享

手机扫一扫分享

编辑 分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

编辑 分享
举报