YOLOv3最全复现代码合集（含PyTorch/TensorFlow和Keras等）

2019 年 12 月 25 日 极市平台

加入极市专业CV交流群，与6000+来自腾讯，华为，百度，北大，清华，中科院等名企名校视觉开发者互动交流！更有机会与李开复老师等大牛群内互动！

同时提供每月大咖直播分享、真实项目需求对接、干货资讯汇总，行业技术交流。关注 极市平台 公众号，回复 加群，立刻申请入群~

前言

很多项目的star数量在2019年都有很大的变化，同时有的库应该还在持续更新，期间也有TensorFlow2的推出，因此这版整理已加入TF2-YOLOv3。

要知道YOLO系列官方源码都是用 C 语言编写的，代码太"硬核"，很多人习惯用Python搞事情，所以网上出现了各种基于 xxx 框架的 YOLOv3复现版本。

本文将基于不同深度学习框架的 YOLOv3 复现代码进行汇总（自认为还不错的复现代码），为了方便各位浏览，下述内容已经同步上传至github：

https://github.com/amusi/YOLO-Reproduce-Summary

此次YOLOv3复现代码合集涵盖 5 种常用深度学习框架：

TensorFlow（新增TensorFlow2）
PyTorch
Keras
Caffe
MXNet

主要信息有：是否支持训练和 star 星数

TensorFlow

https://github.com/YunYang1994/tensorflow-yolov3 支持训练(2070 star)
https://github.com/wizyoung/YOLOv3_TensorFlow 支持训练(978 star)
https://github.com/zzh8829/yolov3-tf2 支持训练(944 star)
https://github.com/mystic123/tensorflow-yolo-v3 不支持训练(687 star)
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

https://github.com/qqwweee/keras-yolo3 支持训练(4916 star)
https://github.com/xiaochus/YOLOv3 不支持训练(517 star)
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

-End-

*延伸阅读

CV细分方向交流群

添加极市小助手微信（ID : cv-mart），备注：研究方向-姓名-学校/公司-城市（如：目标检测-小极-北大-深圳），即可申请加入目标检测、目标跟踪、人脸、工业检测、医学影像、三维&SLAM、图像分割等极市技术交流群（已经添加小助手的好友直接私信），更有每月大咖直播分享、真实项目需求对接、干货资讯汇总，行业技术交流，一起来让思想之光照的更远吧~