【PyTorch】目标检测与场景分割神器 - Detectron2

Backto PyTorch Index

Detectron2 @ 2019-10-10

github $$ docs

Detectron2 ，对 Detectron1 彻底重写，抛弃 Caffe2，直接基于 PyTorch （>= 1.3）, 以 Mask-RCNN benchmark 为基底，全新构建。

• 基于 PyTorch：直观，友好
• 模块化：灵活，可扩展，能够直接在单个或多个 GPU 服务器上进行更快训练。
• Detection All in One: 支持 Box，Mask, Keypoint, Densepose, Semantic segmentation 等
• ModelZoo and Baselines： DensePose, panoptic feature pyramid networks,和Mask RCNN的各种变种等等众多模型
• Detectron2go：新增了将模型产品化部署的软件实现，包括标准的内部数据训练工作流实现、模型压缩量化、模型转化等。

效果惊艳，COCO 厨房

了解代码结构：

• Detectron2源码阅读笔记 (一)Config&Trainer , (二)Registry&build_*方法

一个尝鲜教程： Detectron2 尝鲜 – 训练一个红绿灯检测模型 ， 格式化数据集，给定模型，定义参数，测试可行性，训练，评估，稳！