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

点击上方“CVer”，选择加"星标"置顶

重磅干货，第一时间送达

前言

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

YOLOv4复现下载

在CVer公众号后台回复：YOLO复现，即可下载

重磅！CVer-目标检测 微信交流群已成立

扫码添加CVer助手，可申请加入CVer-目标检测 微信交流群，目前已汇集4100人！涵盖2D/3D目标检测、小目标检测、遥感目标检测等。互相交流，一起进步！

同时也可申请加入CVer大群和细分方向技术群，细分方向已涵盖：目标检测、图像分割、目标跟踪、人脸检测&识别、OCR、姿态估计、超分辨率、SLAM、医疗影像、Re-ID、GAN、NAS、深度估计、自动驾驶、强化学习、车道线检测、模型剪枝&压缩、去噪、去雾、去雨、风格迁移、遥感图像、行为识别、视频理解、图像融合、图像检索、论文投稿&交流、PyTorch和TensorFlow等群。

一定要备注：研究方向+地点+学校/公司+昵称（如目标检测+上海+上交+卡卡），根据格式备注，可更快被通过且邀请进群

▲长按加群

▲长按关注我们

整理不易，请给CVer点赞和在看！