YOLOv3最全復現程式碼合集(含PyTorch/TensorFlow和Keras等)
點選上方“CVer”,選擇“置頂公眾號”
重磅乾貨,第一時間送達
前戲
2018年3月26日,CVer第一時間推文:YOLOv3:你一定不能錯過
2019年3月23日,CVer推文:重磅!YOLOv3最全復現程式碼合集(含TensorFlow/PyTorch和Keras等)
想想距離上次整理已經大半年,很多專案的star數量應該變化很大,而且有的庫應該還在持續更新,期間也有TensorFlow2的推出,所以這版整理已加入TF2-YOLOv3。
要知道YOLO系列官方原始碼都是用 C 語言編寫的,程式碼太"硬核",很多人習慣用Python搞事情,所以網上出現了各種基於 xxx 框架的 YOLOv3復現版本。
Amusi 將基於不同深度學習框架的 YOLOv3 復現程式碼進行彙總(自認為還不錯的復現程式碼),為了方便各位 CVers 瀏覽,下述內容已經同步上傳至github:
https://github.com/amusi/YOLO-Reproduce-Summary
此次YOLOv3復現程式碼合集涵蓋 5 種常用深度學習框架:
TensorFlow(新增TensorFlow2)
PyTorch
Keras
Caffe
MXNet
主要資訊有:是否支援訓練和 star 星數
TensorFlow
1. https://github.com/YunYang1994/tensorflow-yolov3 支援訓練(2070 star)
2. https://github.com/wizyoung/YOLOv3_TensorFlow 支援訓練(978 star)
3. https://github.com/zzh8829/yolov3-tf2 支援訓練(944 star)
4. https://github.com/mystic123/tensorflow-yolo-v3 不支援訓練(687 star)
5. https://github.com/maiminh1996/YOLOv3-tensorflow 支援訓練(276 star)
PyTorch
https://github.com/eriklindernoren/PyTorch-YOLOv3 支援訓練(3316 star)
https://github.com/ultralytics/yolov3 支援訓練(971 star)
這版都有APP版本了
YOLOv3 in PyTorch > ONNX > CoreML > iOS
3. https://github.com/ayooshkathuria/pytorch-yolo-v3 不支援訓練(2400 star)
4. https://github.com/ayooshkathuria/YOLO_v3_tutorial_from_scratch 不支援訓練(1568 star)
5. https://github.com/TencentYoutuResearch/ObjectDetectionOneStageDet/tree/master/yolo 支援訓練(1531 star)
6. https://github.com/BobLiu20/YOLOv3_PyTorch 支援訓練(463 star)
7. https://github.com/DeNA/PyTorch_YOLOv3 支援訓練(280 star)
Keras
1. https://github.com/qqwweee/keras-yolo3 支援訓練(4916 star)
2. https://github.com/xiaochus/YOLOv3 不支援訓練(517 star)
3. https://github.com/Adamdad/keras-YOLOv3-mobilenet 支援訓練(427 star)
Caffe
https://github.com/eric612/MobileNet-YOLO 支援訓練(601 star)
https://github.com/ChenYingpeng/caffe-yolov3 不支援訓練(294 star)
https://github.com/eric612/Caffe-YOLOv3-Windows 支援訓練(167 star)
MXNet
https://github.com/dmlc/gluon-cv/tree/master/gluoncv/model_zoo/yolo 支援訓練(3328 star)
注:上述內容已經同步上傳至 https://github.com/amusi/YOLO-Reproduce-Summary ,大家也可以直接點選"閱讀原文" 進行訪問,歡迎 star!
備註:公眾號選單包含了整理了一本AI小抄,非常適合在通勤路上用學習。
往期精彩回顧
那些年做的學術公益-你不是一個人在戰鬥適合初學者入門人工智慧的路線及資料下載機器學習線上手冊深度學習線上手冊備註:加入本站微信群或者qq群,請回復“加群”加入知識星球(4500+使用者,ID:92416895),請回復“知識星球”
喜歡文章,點個在看
相關文章
- 對比復現34個預訓練模型,PyTorch和Keras你選誰?模型PyTorchKeras
- YOLOv3 的 TensorFlow 實現,GitHub 完整原始碼解析YOLOGithub原始碼
- pytorch實現yolov3(3) 實現forwardPyTorchYOLOForward
- pytorch實現yolov3(1) yolov3基本原理PyTorchYOLO
- [深度學習]人臉檢測-Tensorflow2.x keras程式碼實現深度學習Keras
- CVPR 2020程式碼開源的論文最全合集
- 點贊收藏:PyTorch常用程式碼段整理合集PyTorch
- Tensorflow-kerasKeras
- PyTorch和TensorFlow比較 - thegradientPyTorch
- Linux 下Anaconda 安裝 Tensorflow 和 kerasLinuxKeras
- pytorch和tensorflow的愛恨情仇之一元線性迴歸例子(keras插足啦)PyTorchKeras
- DBA面試資源合集(含Oracle、MySQL、Redis等)面試OracleMySqlRedis
- pytorch實現yolov3(4) 非極大值抑制nmsPyTorchYOLO
- 01_pytorch和tensorflow的區別PyTorch
- 教程帖:使用TensorFlow服務和Flask部署Keras模型!FlaskKeras模型
- 【目標檢測】2萬字詳解 RCNN系列 YOLO系列 YOLOv3程式碼實現全流程詳解 pytorchCNNYOLOPyTorch
- 72 頁 PPT,帶你梳理神經網路完整架構(含 PyTorch 實現程式碼片段)神經網路架構PyTorch
- pytorch實現yolov3(5) 實現端到端的目標檢測PyTorchYOLO
- Tensorflow-keras 理論 & 實戰Keras
- 【tf.keras】tf.keras使用tensorflow中定義的optimizerKeras
- SciTech-BigDataAIML-Tensorflow-Keras是用在Tensorflow的高層APIAIKerasAPI
- TensorFlow 2.0中的tf.keras和Keras有何區別?為什麼以後一定要用tf.keras?Keras
- Laravel常用程式碼合集Laravel
- Tensorflow2的Keras介面:compile、fitKerasCompile
- Keras/Tensorflow選擇GPU/CPU執行KerasGPU
- TensorFlow之keras.layers.Conv2D( )Keras
- 新PyTorch API:幾行程式碼實現不同注意力變體,兼具FlashAttention效能和PyTorch靈活性PyTorchAPI行程
- Java秋招面經大合集(含BAT等大廠面經,均已拿offer)JavaBAT
- pytorch 程式碼出現 ‘segmentation fault (core dump)’ 問題PyTorchSegmentation
- 【Tensorflow_DL_Note12】TensorFlow中LeNet-5模型的實現程式碼模型
- js手寫程式碼合集JS
- 人臉口罩檢測(含執行程式碼+資料集)Pytorch+TensorRT+Xavier NX行程PyTorch
- 如何將tensorflow1.x程式碼改寫為pytorch程式碼(以圖注意力網路(GAT)為例)PyTorch
- TensorFlow2.0教程-使用keras訓練模型Keras模型
- TensorFlow2.0 + CNN + keras + 人臉識別CNNKeras
- Keras 轉換成 Tensorflow 模型格式並使用Keras模型
- 5行程式碼實現微信小程式模版訊息推送 (含推送後臺和小程式原始碼)行程微信小程式原始碼
- 檢視tensorflow,pytorch是否可用GPUPyTorchGPU