YOLOv4最全复现代码合集（含PyTorch/TF/Keras和Caffe等）

前言

2020年4月24日，CVer第一时间推文：大神接棒，YOLOv4来了！

2020年6月28日，CVer第一时间推文：YOLOv4-Tiny来了！371 FPS！

距离YOLOv4正式推出，已经2个多月了。因为YOLO系列官方源码（v1-v4）都是用 C 语言编写的，代码太"硬核"。很多人习惯用Python搞事情，所以网上出现了各种基于 xxx 框架的 YOLOv4复现版本。

其实去年Amusi 就整理过YOLOv3的最全复现合集，详见：YOLOv3最全复现代码合集（含PyTorch/TensorFlow和Keras等）

现在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数量不一定越高越好