OpenMMLab

OpenMMLab 在Github上不是一个单独项目,除了大家所熟知的 Github 上万 star 目标检测库 MMDetection,还有其他方向的代码库和数据集,目前Github总星标超过 1.7 万。是CV方向系统性较强、社区活跃的开源平台。
1. MMCV
MMCV是用于计算机视觉研究的基础Python库,支持OpenMMLab旗下其他开源库。主要功能是I/O、图像视频处理、标注可视化、各种CNN架构、各类CUDA操作算子。
2.MMDetection
MMDetection是基于PyTorch的开源目标检测工具箱。是OpenMMLab最知名的开源库,几乎是研究目标检测必备!
主要特点:
- 模块化设计
 - 支持开箱即用的多方法
 - 高效率
 - SOTA
 
主持的主干网:
- ResNet
 - ResNeXt
 - VGG
 - HRNet
 - RegNet
 - Res2Net
 
支持的算法:
- RPN
 - Fast R-CNN
 - Faster R-CNN
 - Mask R-CNN
 - Cascade R-CNN
 - Cascade Mask R-CNN
 - SSD
 - RetinaNet
 - GHM
 - Mask Scoring R-CNN
 - Double-Head R-CNN
 - Hybrid Task Cascade
 - Libra R-CNN
 - Guided Anchoring
 - FCOS
 - RepPoints
 - Foveabox
 - FreeAnchor
 - NAS-FPN
 - ATSS
 - FSAF
 - PAFPN
 - Dynamic R-CNN
 - PointRend
 - CARAFE
 - DCNv2
 - Group Normalization
 - Weight Standardization
 - OHEM
 - Soft-NMS
 - Generalized Attention
 - GCNet
 - Mixed Precision (FP16) Training
 - InstaBoost
 - GRoIE
 - DetectoRS
 - Generalized Focal Loss
 
3. MMDetection3D
从CVPR2020 中也可以看出3D目标检测研究异常火热,该库是专门用于3D目标检测的开源库。
主要特点:
- 支持开箱即用的多模态/单模态检测器
 - 支持开箱即用的室内/室外检测器
 - 与2D目标检测自然融合
 - 高效率
 

4. MMSegmentation
MMSegmentation是一个基于PyTorch的开源语义分割工具箱.
主要特点:
- 统一基准
 - 模块化设计
 - 支持开箱即用的多方法
 - 高效率
 

支持的骨干网:
- ResNet
 - ResNeXt
 - HRNet
 
支持的算法:
- FCN
 - PSPNet
 - DeepLabV3
 - PSANet
 - DeepLabV3+
 - UPerNet
 - NonLocal Net
 - EncNet
 - CCNet
 - DANet
 - GCNet
 - ANN
 - OCRNet
 
5. MMClassification
MMClassification是基于PyTorch的开源图像分类工具箱。
主要特点:
- 各种骨干与预训练模型
 - Bag of training tricks
 - 大规模训练配置
 - 高效率与可扩展性
 

支持的骨干网:
- ResNet
 - ResNeXt
 - SE-ResNet
 - SE-ResNeXt
 - RegNet
 - ShuffleNetV1
 - ShuffleNetV2
 - MobileNetV2
 - MobileNetV3
 
6. MMPose
MMPose是一个基于PyTorch的开源姿势估计工具箱。

7. MMAction
MMAction是一个基于PyTorch开放源代码的工具箱,用于动作理解。
主要特点:
可以解决以下任务:
从剪辑视频中进行动作识别
未剪辑视频中的时序动作检测(也称为动作定位)
未剪辑视频中的时空动作检测。
支持各种数据集
支持多动作理解框架
模块化设计
8. MMAction2
MMAction2是一个基于PyTorch开放源代码的工具箱,用于动作理解。
主要特点:
- 模块化设计
 - 支持多种数据集
 - 支持多重动作理解框架
 - 完善的测试和记录
 MMAction2比MMAction支持的算法更多,速度更快,开发者也更活跃。

支持的动作识别算法:
- TSN
 - TSM
 - R(2+1)D
 - I3D
 - SlowOnly
 - SlowFast
 
支持的动作定位算法:
- BMN
 - BSN
 
9. MMSkeleton
MMSkeleton
用于人体姿势估计,基于骨架的动作识别和动作合成。
特点:
- 高扩展性
 - 多任务
 

10. MMFashion
MMFashion是一个基于PyTorch的开源视觉时尚分析工具箱。
特点:
- 灵活:模块化设计,易于扩展
 - 友好:外行用户的现成模型
 - 全面:支持各种时装分析任务
 支持应用:
- 服饰属性预测
 - 服饰识别与检索
 - 服饰特征点检测
 - 服饰解析和分割
 - 服饰搭配推荐
 

11. MMEditing
MMEditing是基于PyTorch的开源图像和视频编辑工具箱
主要特点:
- 模块化设计
 - 在编辑中支持多任务
 - SOTA
 

12.OpenPCDet
OpenPCDet 是一个清晰,简单,自成体系的开源项目,用于基于LiDAR的3D目标检测。
