mmdetection基于 PyTorch 的對象檢測工具箱
mmdetection 是一個基于 PyTorch 的開源對象檢測工具箱,它提供了已公開發(fā)表的多種視覺檢測核心模塊,通過這些模塊的組合,可以迅速搭建出各種著名的檢測框架。
主要特性:
-
模塊化設計
通過組合不同的組件,可以容易地構(gòu)建定制的對象檢測框架。
-
支持多個框架開箱即用
該工具箱直接支持流行的檢測框架,如 Faster RCNN、Mask RCNN、RetinaNet 等。
-
高效
所有基本的 bbox 和掩碼操作現(xiàn)在都在 GPU 上運行。不同模型上,訓練速度比 Detectron 快約 5%~20%。
-
最先進的
重構(gòu)自 MMDet 團隊的代碼庫,而 MMDet 贏得了 COCO Detection 2018 挑戰(zhàn)賽。
評論
圖片
表情
