腾讯优图:开源YOLO系列代码(含YOLOv3以及各种backbone)
- 2019 年 12 月 31 日
- 筆記
本文原定于上周跟大家见面,但因不可抗拒原因"被"推迟一周。在此感谢腾讯优图实验室开源这么棒的项目,并特别鸣谢优图小伙伴:mileistone(YOLO系列复现作者)。
前言
本文给各位CVers介绍一个由腾讯优图实验室开源的One-Stage目标检测项目。链接如下:
https://github.com/TencentYoutuResearch/ObjectDetection-OneStageDet
前不久,刚刚push上YOLO系列代码。来看一下阵容:
基于PyTorch的YOLO系列代码实现,包含Tiny-YOLOv2、YOLOv2、Tiny-YOLOv3、YOLO-v3以及MobileNet、MobileNetv2、ShuffleNet、ShuffleNetv2、SqueezeNext、Xception等backbone。
这阵容有点…还是基于PyTorch的…不说了,star,fork,watch走起!
腾讯优图—YOLO
这里简单介绍项目里的部分亮点
环境
- Python 3.6
- PyTorch 0.4.0
性能
哇,这mAP!哇,这速度!
支持训练&评估
之前很多同学向我吐槽,网上可以找到基于PyTorch的YOLOv3代码,但只支持inference,不能训练。那你看看腾讯优图这个,不仅教你怎么训练,还手把手教你评估。远程比心!
预训练权重
代码有了,教程有了,再看看预训练权重。
不仅仅有Google Drive,还有良心的百度云链接(国内coding man福利)
PS:googledrieve打错了,大家快去提交issue,还能成为优图开源项目的贡献者(手动狗头.jpg)
如果你想学PyTorch,你还想学目标检测(如YOLO),那么这个开源项目非你莫属!不说了,俺去download撸代码了
开源代码链接:
https://github.com/TencentYoutuResearch/ObjectDetection-OneStageDet/tree/master/yolo
作者(mileistone)知乎链接:
https://zhuanlan.zhihu.com/p/54959146