天天看點

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