天天看点

yolov4源码_【CV】YOLOv4最全复现代码合集(含PyTorch/TF/Keras和Caffe等)

前言

2020年4月24日,CVer第一时间推文:大神接棒,YOLOv4来了!

2020年6月28日,CVer第一时间推文:YOLOv4-Tiny来了!371 FPS!

yolov4源码_【CV】YOLOv4最全复现代码合集(含PyTorch/TF/Keras和Caffe等)

距离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

yolov4源码_【CV】YOLOv4最全复现代码合集(含PyTorch/TF/Keras和Caffe等)

1. https://github.com/Tianxiaomo/pytorch-YOLOv4 支持训练(926 star)

2. https://github.com/bubbliiiing/yolov4-pytorch 支持训练(120 star)

TensorFlow

yolov4源码_【CV】YOLOv4最全复现代码合集(含PyTorch/TF/Keras和Caffe等)

1. https://github.com/hunglc007/tensorflow-yolov4-tflite 支持训练(464 star)

2. https://github.com/rrddcc/YOLOv4_tensorflow 支持训练(116 star)

yolov4源码_【CV】YOLOv4最全复现代码合集(含PyTorch/TF/Keras和Caffe等)

tensorflow-yolov4-tflite

Keras

yolov4源码_【CV】YOLOv4最全复现代码合集(含PyTorch/TF/Keras和Caffe等)

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)

yolov4源码_【CV】YOLOv4最全复现代码合集(含PyTorch/TF/Keras和Caffe等)

Keras-YOLOv4

PaddlePaddle

yolov4源码_【CV】YOLOv4最全复现代码合集(含PyTorch/TF/Keras和Caffe等)

1. https://github.com/miemie2013/Paddle-YOLOv4 支持训练(14 star)

Caffe

yolov4源码_【CV】YOLOv4最全复现代码合集(含PyTorch/TF/Keras和Caffe等)

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

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

TensorRT

yolov4源码_【CV】YOLOv4最全复现代码合集(含PyTorch/TF/Keras和Caffe等)

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

tkDNN

yolov4源码_【CV】YOLOv4最全复现代码合集(含PyTorch/TF/Keras和Caffe等)

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

注1:上述内容已经同步上传至github:https://github.com/amusi/YOLO-Reproduce-Summary

注2:要额外说明一下,YOLOv4中的tricks非常多,工作量相当大,复现起来难度也超级大,目前Amusi还没有看到完全复现YOLOv4的项目,希望上述项目加油不断迭代!

注3:项目的star数量不一定越高越好

yolov4源码_【CV】YOLOv4最全复现代码合集(含PyTorch/TF/Keras和Caffe等)

往期精彩回顾

适合初学者入门人工智能的路线及资料下载

机器学习及深度学习笔记等资料打印

机器学习在线手册

深度学习笔记专辑

《统计学习方法》的代码复现专辑

AI基础下载

机器学习的数学基础专辑

获取一折本站知识星球优惠券,复制链接直接打开:

https://t.zsxq.com/yFQV7am

本站qq群1003271085。

加入微信群请扫码进群:

yolov4源码_【CV】YOLOv4最全复现代码合集(含PyTorch/TF/Keras和Caffe等)