今天看啥  ›  专栏  ›  机器学习研究组订阅

资源 | 一个基于PyTorch的目标检测工具箱,商汤联合港中文开源mmdetection

机器学习研究组订阅  · 公众号  · AI  · 2018-10-17 20:11
近日,商汤和港中文联合开源了 mmdetection,这是一个基于 PyTorch 的开源目标检测工具包,属于香港中文大学多媒体实验室 open-mmlab 项目的一部分。该工具包支持 Mask RCNN 等多种流行的检测框架,读者可在 PyTorch 环境下测试不同的预训练模型及训练新的检测分割模型。项目地址:https://github.com/open-mmlab/mmdetection根据 GitHub 项目所述,mmdetection 的主要特征可以总结为以下几个方面:模块化设计:你可以通过连接不同组件轻松构建自定义目标检测框架。支持多个框架,开箱即用:该工具包直接支持多种流行的检测框架,如 Faster RCNN、Mask RCNN、RetinaNet 等。高效:所有基础边界框和掩码运算都在 GPU 上运行。不同模型的训练速度大约比 FAIR 的 Detectron 快 5% ~ 20%。当前最优:这是 MMDet ………………………………

原文地址:访问原文地址
快照地址: 访问文章快照