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

- https://github.com/eriklindernoren/PyTorch-YOLOv3 支援訓練(3316 star)
- 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

- https://github.com/eric612/MobileNet-YOLO 支援訓練(601 star)
- https://github.com/ChenYingpeng/caffe-yolov3 不支援訓練(294 star)
- https://github.com/eric612/Caffe-YOLOv3-Windows 支援訓練(167 star)
MXNet

- https://github.com/dmlc/gluon-cv/tree/master/gluoncv/model_zoo/yolo 支援訓練(3328 star)
註:上述內容已經同步上傳至 https://github.com/amusi/YOLO-Reproduce-Summary
