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

  • 2019 年 12 月 24 日
  • 筆記

前戲

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

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

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

這版都有APP版本了

YOLOv3 in PyTorch > ONNX > CoreML > iOS

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

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

  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

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

註:上述內容已經同步上傳至 https://github.com/amusi/YOLO-Reproduce-Summary