mmdetection商汤科技和香港中文大学最近开源了一个基于Pytorch实现的深度学习目标检测工具箱mmdetection,支持Faster-RCNN,Mask-RCNN,Fast-RCNN等主流的目标检测框架,后续会加入Cascade-RCNN以及其他一系列目标检测框架。选择学习这个框架,是应为它支持的目标检测的网络模型比较多。想详细了解可以阅读github的工程和arxiv的文档 《MMDetection: Open MMLab Detection Toolbox and Benchmark 》
github:https://github.com/open-mmlab/mmdetection
1.支持的算法
支持的一阶段和二阶段目标检测算法还相对齐全
2.框架结构
主要分成backbone , Neck, DenseHead RoIHead这四个主要部分。
3.train pipeline