YOLOv3最全復現程式碼合集(含PyTorch/TensorFlow和Keras等)

風度78發表於2019-12-21

點選上方“CVer”,選擇“置頂公眾號”

重磅乾貨,第一時間送達

YOLOv3最全復現程式碼合集(含PyTorch/TensorFlow和Keras等)

前戲

2018年3月26日,CVer第一時間推文:YOLOv3:你一定不能錯過

2019年3月23日,CVer推文:重磅!YOLOv3最全復現程式碼合集(含TensorFlow/PyTorch和Keras等)

想想距離上次整理已經大半年,很多專案的star數量應該變化很大,而且有的庫應該還在持續更新,期間也有TensorFlow2的推出,所以這版整理已加入TF2-YOLOv3。

要知道YOLO系列官方原始碼都是用 C 語言編寫的,程式碼太"硬核",很多人習慣用Python搞事情,所以網上出現了各種基於 xxx 框架的 YOLOv3復現版本。

Amusi 將基於不同深度學習框架的 YOLOv3 復現程式碼進行彙總(自認為還不錯的復現程式碼),為了方便各位 CVers 瀏覽,下述內容已經同步上傳至github:

https://github.com/amusi/YOLO-Reproduce-Summary

此次YOLOv3復現程式碼合集涵蓋 5 種常用深度學習框架:

  • TensorFlow(新增TensorFlow2)

  • PyTorch

  • Keras

  • Caffe

  • MXNet

主要資訊有:是否支援訓練和 star 星數

TensorFlow

YOLOv3最全復現程式碼合集(含PyTorch/TensorFlow和Keras等)

1. https://github.com/YunYang1994/tensorflow-yolov3 支援訓練(2070 star)

2. https://github.com/wizyoung/YOLOv3_TensorFlow 支援訓練(978 star)

3. https://github.com/zzh8829/yolov3-tf2 支援訓練(944 star)

4. https://github.com/mystic123/tensorflow-yolo-v3 不支援訓練(687 star)

5. https://github.com/maiminh1996/YOLOv3-tensorflow 支援訓練(276 star)

PyTorch

YOLOv3最全復現程式碼合集(含PyTorch/TensorFlow和Keras等)

  1. https://github.com/eriklindernoren/PyTorch-YOLOv3 支援訓練(3316 star)

  2. https://github.com/ultralytics/yolov3 支援訓練(971 star)

這版都有APP版本了

YOLOv3 in PyTorch > ONNX > CoreML > iOS

YOLOv3最全復現程式碼合集(含PyTorch/TensorFlow和Keras等)

3. https://github.com/ayooshkathuria/pytorch-yolo-v3 不支援訓練(2400 star)

4. https://github.com/ayooshkathuria/YOLO_v3_tutorial_from_scratch 不支援訓練(1568 star)

5. https://github.com/TencentYoutuResearch/ObjectDetectionOneStageDet/tree/master/yolo 支援訓練(1531 star)

6. https://github.com/BobLiu20/YOLOv3_PyTorch 支援訓練(463 star)

7. https://github.com/DeNA/PyTorch_YOLOv3 支援訓練(280 star)

Keras

YOLOv3最全復現程式碼合集(含PyTorch/TensorFlow和Keras等)

1. https://github.com/qqwweee/keras-yolo3 支援訓練(4916 star)

2. https://github.com/xiaochus/YOLOv3 不支援訓練(517 star)

3. https://github.com/Adamdad/keras-YOLOv3-mobilenet 支援訓練(427 star)

Caffe

YOLOv3最全復現程式碼合集(含PyTorch/TensorFlow和Keras等)

  1. https://github.com/eric612/MobileNet-YOLO 支援訓練(601 star)

  2. https://github.com/ChenYingpeng/caffe-yolov3 不支援訓練(294 star)

  3. https://github.com/eric612/Caffe-YOLOv3-Windows 支援訓練(167 star)

MXNet

YOLOv3最全復現程式碼合集(含PyTorch/TensorFlow和Keras等)

  1. https://github.com/dmlc/gluon-cv/tree/master/gluoncv/model_zoo/yolo 支援訓練(3328 star)

注:上述內容已經同步上傳至 https://github.com/amusi/YOLO-Reproduce-Summary ,大家也可以直接點選"閱讀原文" 進行訪問,歡迎 star!

YOLOv3最全復現程式碼合集(含PyTorch/TensorFlow和Keras等)

備註:公眾號選單包含了整理了一本AI小抄非常適合在通勤路上用學習

YOLOv3最全復現程式碼合集(含PyTorch/TensorFlow和Keras等)

往期精彩回顧

那些年做的學術公益-你不是一個人在戰鬥適合初學者入門人工智慧的路線及資料下載機器學習線上手冊深度學習線上手冊備註:加入本站微信群或者qq群,請回復“加群”加入知識星球(4500+使用者,ID:92416895),請回復“知識星球”

喜歡文章,點個在看YOLOv3最全復現程式碼合集(含PyTorch/TensorFlow和Keras等)

相關文章