Amusi 将基于不同深度学习框架的 YOLOv3 复现代码进行汇总(自认为还不错的复现代码),为了方便各位 CVers 浏览,下述内容已经同步上传至github:
https://github.com/amusi/YOLO-Reproduce-Summary 此次YOLOv3复现代码合集涵盖 5 种常用深度学习框架:
TensorFlow PyTorch Keras Caffe MXNet
主要信息有:是否支持训练和 star 星数 TensorFlow
https://github.com/YunYang1994/tensorflow-yolov3 支持训练(514 star) https://github.com/mystic123/tensorflow-yolo-v3 不支持训练(446 star) https://github.com/maiminh1996/YOLOv3-tensorflow 支持训练(246 star)
PyTorch
https://github.com/ayooshkathuria/pytorch-yolo-v3 不支持训练(1.5k star) https://github.com/eriklindernoren/PyTorch-YOLOv3 支持训练(1.3k star) https://github.com/ayooshkathuria/YOLO_v3_tutorial_from_scratch 不支持训练(1k star) https://github.com/TencentYoutuResearch/ObjectDetectionOneStageDet/tree/master/yolo 支持训练(976 star)
https://github.com/ultralytics/yolov3 支持训练(971 star)
这版都有APP版本了
https://github.com/BobLiu20/YOLOv3_PyTorch 支持训练(341 star) https://github.com/DeNA/PyTorch_YOLOv3 支持训练(179 star)
Keras
https://github.com/qqwweee/keras-yolo3 支持训练(2.8k star) https://github.com/xiaochus/YOLOv3 不支持训练(418 star) https://github.com/Adamdad/keras-YOLOv3-mobilenet 支持训练(191 star)
Caffe
https://github.com/eric612/MobileNet-YOLO 支持训练(301 star) https://github.com/ChenYingpeng/caffe-yolov3 不支持训练(150 star) https://github.com/eric612/Caffe-YOLOv3-Windows 支持训练(97 star)
MXNet
注:上述内容已经同步上传至 https://github.com/amusi/YOLO-Reproduce-Summary
|