Facebook 釋出 Detectron2:基於 PyTorch 的新一代目標檢測工具

AIBigbull2050發表於2019-10-16

Detectron 是 FAIR 在 2018 年初公開的目標檢測平臺,包含了大量業內最具代表性的目標檢測、影像分割、關鍵點檢測演算法,該框架主要基於 python 和 caffe2 實現,開源專案已獲得了超 2.2w 的星標數。

而 Detectron2 則是剛推出的新版 PyTorch 1.3  (詳情檢視: https://www.leiphone.com/news/201910/9J4PTd833WMwdVoQ.html中一重大新工具,它源於 maskrcnn 基準測試,但它卻是對先前版本 detectron 的一次徹底重寫。Detectron2 包含了更大的靈活性與擴充套件性,並增強了可維護性和可伸縮性,以支援在生產中的用例。目前已在 GitHub 上開源,雷鋒網 AI 開發者將其重要內容整理編譯如下。

Facebook 釋出 Detectron2:基於 PyTorch 的新一代目標檢測工具

Detectron2 簡介

據官方介紹,Detectron2 是 FAIR 的下一代目標檢測和分割研究平臺。目前,它實現了最先進的目標檢測演算法。

Detectron 2 與之前的 detectron 不同,它並非在原版基礎上進行改寫,而是徹底重寫的全新版本。新版目標檢測平臺 Detectron 2 則基於 PyTorch 框架,以 maskrcnn 基準測試作為起點。通過全新的模組化設計,Detectron2 變得更靈活且易於擴充套件,它能夠在單個或多個 GPU 伺服器上提供更快速的訓練速度。

Facebook 釋出 Detectron2:基於 PyTorch 的新一代目標檢測工具

Detectron2 的新特性

  • 基於 PyTorch 框  與 Detectron 在 Caffe2 中實現不同,Detectron2 則基於 PyTorch 實現。PyTorch 提供了一個更直觀的指令式程式設計模型,它允許研究人員和工程師可以更快地迭代模型設計和實驗,並讓開發者受益於 PyTorch 深度學習方法以及不斷改進的 PyTorch 活躍社群。

Facebook 釋出 Detectron2:基於 PyTorch 的新一代目標檢測工具

  • 模組化、可擴充套件設計 在 Detectron2 中,引入了模組化設計,使用者可以將自定義模組插入目標檢測系統的任意部分。這意味著許多新的研究專案可以用數百行程式碼編寫,而核心 Detectron2 庫和全新的研究之間可以實現完全分離。

    通過建立新模型並發現新方法,可以不斷完善模組化和可擴充套件的設計,這也幫助 Detectron2 變得更加靈活。其模組化特性也能夠使其更有效地幫助研究人員探索最先進的演算法設計。

  • 新模型和新功能 Detectron2 包括原始 Detectron 中的所有模型,例如 Faster R-CNN,Mask R-CNN,RetinaNet 和 DensePose。除此之外,Detectron2 還加入了幾個新模型,包括:Cascade R-CNN,Panoptic FPN 和 TensorMask,未來還將繼續新增更多演算法。

    在 Detectron2 還包含了一些新功能,例如:同步批處理規範化(synchronous Batch Norm),以及對 LVIS 等新資料集的支援等。 

Facebook 釋出 Detectron2:基於 PyTorch 的新一代目標檢測工具

  • 新任務支援 Detectron2 支援與目標檢測有關的一系列任務。和之前的 Detectron 一樣,它也支援目標檢測、例項分割、人體姿態估計等任務。此外,Detectron2 還支援語義分割和全景分割。

  • 高實現質量 重寫的 Detectron2 能夠重新檢測底層的設計決策,並解決原來 Detectron 的幾個實現問題。 

  • 速度和可擴充套件性 通過將整個訓練 pipeline 移至 GPU,能夠使 Detectron2 比原來的 Detectron 更快,而且可以更加方便在各種標準型號下進行 GPU 伺服器的分散式訓練,從而輕鬆擴充套件訓練資料集。

  • Detectron2go 新增了將模型產品化部署的軟體實現,包括:標準的內部資料訓練工作流實現、模型壓縮量化、模型轉化等。

更多關於Detectron2 :

http://ai.facebook.com/blog/-detectron2-a-pytorch-based-modular-object-detection-library-/  

GitHub 開源地址:

https://github.com/facebookresearch/detectron2  



https://www.leiphone.com/news/201910/P2sq3UjmbsI6QdiW.html



來自 “ ITPUB部落格 ” ,連結:http://blog.itpub.net/69946223/viewspace-2660110/,如需轉載,請註明出處,否則將追究法律責任。

相關文章