OpenMMLab 为香港中文大学-商汤科技联合实验室 MMLab 开源的算法平台,不到两年时间,已经包含众多 SOTA 计算机视觉算法。
OpenMMLab 在Github上不是一个单独项目,除了大家所熟知的 Github 上万 star 目标检测库 MMDetection,还有其他方向的代码库和数据集,非常值得从事计算机视觉研发的朋友关注。
近期 OpenMMLab 进行了密集更新,新增了多个库,官方称涉及超过 10 个研究方向,开放超过 100 种算法和 600 种预训练模型,目前Github总星标超过 1.7 万。是CV方向系统性较强、社区活跃的开源平台。
这些库大部分都基于深度学习 PyTorch 框架,算法紧跟前沿,方便易用,文档较为丰富,无论对于研究还是工程开发的朋友都很值得了解。
项目主页:
http://openmmlab.org/
本文带领大家细数各个代码库,相信总有一款适合你!
MMCV
MMCV是用于计算机视觉研究的基础Python库,支持OpenMMLab旗下其他开源库。
Github | https://github.com/open-mmlab/mmcv
主要功能是I/O、图像视频处理、标注可视化、各种CNN架构、各类CUDA操作算子。
MMDetection
MMDetection是基于PyTorch的开源目标检测工具箱。是OpenMMLab最知名的开源库,几乎是研究目标检测必备!
主要特点:
模块化设计
支持开箱即用的多方法
高效率
SOTA
demo:
Github | https://github.com/open-mmlab/mmdetection
主持的主干网:
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
论文盘点 | CVPR 2020 -目标检测篇
MMDetection3D
从CVPR2020 中也可以看出3D目标检测研究异常火热,该库是专门用于3D目标检测的开源库。
主要特点:
支持开箱即用的多模态/单模态检测器
支持开箱即用的室内/室外检测器