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

450 阅读3分钟

点击上方“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:

github.com/amusi/YOLO-…

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

  • TensorFlow(新增TensorFlow2)
  • PyTorch
  • Keras
  • Caffe
  • MXNet

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

TensorFlow

1. github.com/YunYang1994… 支持训练(2070 star)

2. github.com/wizyoung/YO… 支持训练(978 star)

3. github.com/zzh8829/yol…

4. github.com/mystic123/t… 不支持训练(687 star)

5. github.com/maiminh1996… 支持训练(276 star)

PyTorch

  1. github.com/eriklindern… 支持训练(3316 star)
  2. github.com/ultralytics… 支持训练(971 star)

这版都有APP版本了

YOLOv3 in PyTorch > ONNX > CoreML > iOS

3. github.com/ayooshkathu… 不支持训练(2400 star)

4. github.com/ayooshkathu… 不支持训练(1568 star)

5. github.com/TencentYout… 支持训练(1531 star)

6. github.com/BobLiu20/YO… 支持训练(463 star)

7. github.com/DeNA/PyTorc… 支持训练(280 star)

Keras

1. github.com/qqwweee/ker… 支持训练(4916 star)

2. github.com/xiaochus/YO… 不支持训练(517 star)

3. github.com/Adamdad/ker… 支持训练(427 star)

Caffe

  1. github.com/eric612/Mob… 支持训练(601 star)
  2. github.com/ChenYingpen… 不支持训练(294 star)
  3. github.com/eric612/Caf… 支持训练(167 star)

MXNet

  1. github.com/dmlc/gluon-… 支持训练(3328 star)

注:上述内容已经同步上传至 github.com/amusi/YOLO-… ,大家也可以直接点击"阅读原文" 进行访问,欢迎 star!

\

备注:公众号菜单包含了整理了一本 AI小抄非常适合在通勤路上用学习

往期精彩回顾

那些年做的学术公益-你不是一个人在战斗适合初学者入门人工智能的路线及资料下载机器学习在线手册深度学习在线手册备注:加入本站微信群或者qq群,请回复“加群”加入知识星球(4500+用户,ID:92416895),请回复“知识星球”

喜欢文章,点个 在看