47

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

 4 years ago
source link: http://mp.weixin.qq.com/s?__biz=MzI5MDUyMDIxNA%3D%3D&%3Bmid=2247492790&%3Bidx=2&%3Bsn=b9df4965444048ea42811e7d66dacbc1
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.
neoserver,ios ssh client

加入极市 专业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

auAFb2j.png!web

  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

2636v2q.png!web

  1. https://github.com/eriklindernoren/PyTorch-YOLOv3 支持训练(3316 star)

  2. https://github.com/ultralytics/yolov3 支持训练(971 star)

这版都有APP版本了

YOLOv3 in PyTorch > ONNX > CoreML > iOS

RRRRBby.jpg!web

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

ERZJFzi.png!web

  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

BJZ7nai.png!web

  1. https://github.com/eric612/MobileNet-YOLO 支持训练(601 star)

  2. https://github.com/ChenYingpeng/caffe-yolov3 不支持训练(294 star)

  3. https://github.com/eric612/Caffe-YOLOv3-Windows 支持训练(167 star)

MXNet

imQZnm3.png!web

  1. 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、图像分割等极市技术交流群 (已经添加小助手的好友直接私信) ,更有每月 大咖直播分享、真实项目需求对接、干货资讯汇总,行业技术交流 一起来让思想之光照的更远吧~

RrYj22I.jpg!web

△长按添加极市小助手

Yjqyyiq.jpg!web

△长按关注极市平台

觉得有用麻烦给个在看啦~    uE7RJjy.gif


About Joyk


Aggregate valuable and interesting links.
Joyk means Joy of geeK