騰訊優圖:開源YOLO系列程式碼(含YOLOv3以及各種backbone)
- 2019 年 12 月 31 日
- 筆記
本文原定於上周跟大家見面,但因不可抗拒原因"被"推遲一周。在此感謝騰訊優圖實驗室開源這麼棒的項目,並特別鳴謝優圖小夥伴:mileistone(YOLO系列復現作者)。
前言
本文給各位CVers介紹一個由騰訊優圖實驗室開源的One-Stage目標檢測項目。鏈接如下:
https://github.com/TencentYoutuResearch/ObjectDetection-OneStageDet
前不久,剛剛push上YOLO系列程式碼。來看一下陣容:
基於PyTorch的YOLO系列程式碼實現,包含Tiny-YOLOv2、YOLOv2、Tiny-YOLOv3、YOLO-v3以及MobileNet、MobileNetv2、ShuffleNet、ShuffleNetv2、SqueezeNext、Xception等backbone。

這陣容有點…還是基於PyTorch的…不說了,star,fork,watch走起!
騰訊優圖—YOLO
這裡簡單介紹項目里的部分亮點
環境
- Python 3.6
- PyTorch 0.4.0
性能
哇,這mAP!哇,這速度!


支援訓練&評估
之前很多同學向我吐槽,網上可以找到基於PyTorch的YOLOv3程式碼,但只支援inference,不能訓練。那你看看騰訊優圖這個,不僅教你怎麼訓練,還手把手教你評估。遠程比心!


預訓練權重
程式碼有了,教程有了,再看看預訓練權重。
不僅僅有Google Drive,還有良心的百度雲鏈接(中國coding man福利)
PS:googledrieve打錯了,大家快去提交issue,還能成為優圖開源項目的貢獻者(手動狗頭.jpg)

如果你想學PyTorch,你還想學目標檢測(如YOLO),那麼這個開源項目非你莫屬!不說了,俺去download擼程式碼了
開源程式碼鏈接:
https://github.com/TencentYoutuResearch/ObjectDetection-OneStageDet/tree/master/yolo
作者(mileistone)知乎鏈接:
https://zhuanlan.zhihu.com/p/54959146