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

前言

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

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

【CV】YOLOv4最全复现代码合集(含PyTorch/TF/Keras和Caffe等)_第1张图片

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

【CV】YOLOv4最全复现代码合集(含PyTorch/TF/Keras和Caffe等)_第2张图片

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

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

TensorFlow

【CV】YOLOv4最全复现代码合集(含PyTorch/TF/Keras和Caffe等)_第3张图片

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

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

tensorflow-yolov4-tflite

Keras

【CV】YOLOv4最全复现代码合集(含PyTorch/TF/Keras和Caffe等)_第4张图片

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)

【CV】YOLOv4最全复现代码合集(含PyTorch/TF/Keras和Caffe等)_第5张图片

Keras-YOLOv4

PaddlePaddle

【CV】YOLOv4最全复现代码合集(含PyTorch/TF/Keras和Caffe等)_第6张图片

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

Caffe

【CV】YOLOv4最全复现代码合集(含PyTorch/TF/Keras和Caffe等)_第7张图片

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

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

TensorRT

【CV】YOLOv4最全复现代码合集(含PyTorch/TF/Keras和Caffe等)_第8张图片

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

tkDNN

【CV】YOLOv4最全复现代码合集(含PyTorch/TF/Keras和Caffe等)_第9张图片

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等))