前言
2020年4月24日,CVer第一时间推文:大神接棒,YOLOv4来了!
2020年6月28日,CVer第一时间推文:YOLOv4-Tiny来了!371 FPS!
![](https://img.laitimes.com/img/__Qf2AjLwojIjJCLyojI0JCLicmbw5SNxQzNkdzNmJmZ0Y2NzMDM2ATY3EzYjdjZ0MjY4QTZ08CX0JXZ252bj91Ztl2Lc52YucWbp5GZzNmLn9Gbi1yZtl2Lc9CX6MHc0RHaiojIsJye.png)
距离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
注2:要额外说明一下,YOLOv4中的tricks非常多,工作量相当大,复现起来难度也超级大,目前Amusi还没有看到完全复现YOLOv4的项目,希望上述项目加油不断迭代!
注3:项目的star数量不一定越高越好
往期精彩回顾
适合初学者入门人工智能的路线及资料下载
机器学习及深度学习笔记等资料打印
机器学习在线手册
深度学习笔记专辑
《统计学习方法》的代码复现专辑
AI基础下载
机器学习的数学基础专辑
获取一折本站知识星球优惠券,复制链接直接打开:
https://t.zsxq.com/yFQV7am
本站qq群1003271085。
加入微信群请扫码进群:
yolov4源码_【CV】YOLOv4最全复现代码合集(含PyTorch/TF/Keras和Caffe等)