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