前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >YOLOv3最全复现代码合集(含PyTorch/TensorFlow和Keras等)

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

作者头像
Amusi
发布2019-12-24 16:28:31
1.3K0
发布2019-12-24 16:28:31
举报
文章被收录于专栏:CVerCVer

前戏

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

  1. https://github.com/eriklindernoren/PyTorch-YOLOv3 支持训练(3316 star)
  2. 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

  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

  1. https://github.com/dmlc/gluon-cv/tree/master/gluoncv/model_zoo/yolo 支持训练(3328 star)

注:上述内容已经同步上传至 https://github.com/amusi/YOLO-Reproduce-Summary

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2019-12-20,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 CVer 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 前戏
  • TensorFlow
  • PyTorch
  • Keras
  • Caffe
  • MXNet
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档