MMDetection是基於深度學習的目標檢測算法框架,使用MMDetection能快速搭建目標檢測網絡,實現目標檢測。本文將介紹MMDetection:
MMDetection,包含一組豐富的對象檢測和實例細分方法以及相關的組件和模塊。它不僅包括訓練和推理代碼,還為200多個網絡模型提供權重。該工具箱是迄今為止最完整的檢測工具箱。
主要特點
模塊化設計我們將檢測框架分解為不同的組件,並且可以通過組合不同的模塊輕鬆構建定製的對象檢測框架。開箱即用地支持多種框架該工具箱直接支持流行和現代的檢測框架,例如 更快的RCNN,Mask RCNN,RetinaNet等高效率所有基本的bbox和mask操作都在GPU上運行。 訓練速度比其他代碼庫(包括Detectron2,maskrcnn-benchmark和SimpleDet)快或可比。最先進的該工具箱源自MMDet團隊開發的代碼庫,該團隊在2018年贏得了COCO檢測挑戰賽的冠軍,我們一直在推動它向前發展。除了MMDetection之外,我們還發布了mmcv庫用於計算機視覺研究,該庫在很大程度上依賴於此工具箱。更多細節請關注github MMDetection