前言

2020年4月24日，CVer第一時間推文：大神接棒，YOLOv4來了！

2020年6月28日，CVer第一時間推文：YOLOv4-Tiny來了！371 FPS！

距離YOLOv4正式推出，已經2個多月了。因為YOLO系列官方源碼（v1-v4）都是用 C 語言編寫的，代碼太"硬核"。很多人習慣用Python搞事情，所以網上出現了各種基于 xxx 框架的 YOLOv4復現版本。

現在Amusi 將基于不同深度學習框架的 YOLOv4 復現代碼進行匯總（自認為還不錯的復現代碼），為了方便各位 CVers 瀏覽，下述內容已經同步上傳至github：

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

此次YOLOv4 復現代碼合集涵蓋下面幾種常用深度學習框架（含推理）：

• PyTorch

• TensorFlow

• Keras

• PaddlePaddle

• Caffe

• TensorRT

• tkDNN
  

主要信息有：是否支持訓練和 star 星數

PyTorch

1. https://github.com/Tianxiaomo/pytorch-YOLOv4 支持訓練(926?star)

2. https://github.com/bubbliiiing/yolov4-pytorch 支持訓練(120?star)

TensorFlow

1. https://github.com/hunglc007/tensorflow-yolov4-tflite 支持訓練(464?star)

2. https://github.com/rrddcc/YOLOv4_tensorflow 支持訓練(116?star)

tensorflow-yolov4-tflite

Keras

1. https://github.com/david8862/keras-YOLOv3-model-set 支持訓練(279?star)

2. https://github.com/miemie2013/Keras-YOLOv4 支持訓練(98 star)

3. https://github.com/bubbliiiing/yolov4-keras 支持訓練(142?star)

4. https://github.com/Ma-Dan/keras-yolo4 支持訓練(220 star)

Keras-YOLOv4

PaddlePaddle

1. https://github.com/miemie2013/Paddle-YOLOv4 支持訓練(14?star)

Caffe

1. https://github.com/ChenYingpeng/caffe-yolov3 (362 star)

2. https://github.com/ChenYingpeng/darknet2caffe (54?star)

TensorRT

1. https://github.com/CaoWGG/TensorRT-YOLOv4 (106?star)

tkDNN

1. https://github.com/ceccocats/tkDNN (177?star)

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

注2：要額外說明一下，YOLOv4中的tricks非常多，工作量相當大，復現起來難度也超級大，目前Amusi還沒有看到完全復現YOLOv4的項目，希望上述項目加油不斷迭代！

注3：項目的star數量不一定越高越好

往期精彩回顧適合初學者入門人工智能的路線及資料下載機器學習及深度學習筆記等資料打印機器學習在線手冊深度學習筆記專輯《統計學習方法》的代碼復現專輯 AI基礎下載機器學習的數學基礎專輯獲取一折本站知識星球優惠券，復制鏈接直接打開：https://t.zsxq.com/yFQV7am本站qq群1003271085。加入微信群請掃碼進群：

總結

以上是生活随笔為你收集整理的【CV】YOLOv4最全复现代码合集（含PyTorch/TF/Keras和Caffe等）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。