方栗子 发自 凹非寺
量子位 报道
来自YOLOv3原作者
YOLOv3,快如闪电,可称目标检测之光。
PyTorch实现教程去年4月就出现了,TensorFlow实现一直零零星星。
现在,有位热心公益的程序猿 (Yunyang1994) ,为它做了纯TensorFlow代码实现。
这份实现,支持用自己的数据训练模型。
介绍一下
TensorFlow实现,包含了以下部分:· YOLOv3架构
· 权重转换器 (Weight Converter)
· 基础版Demo
· GPU和CPU上都支持非极大抑制 (Non-Maximum Suppression)
· 训练pipeline
· COCO mAP计算
来自YOLOv3原作者
快速开始
四个步骤,速速上车。
1.复制这个文件:1$ git clone https://github.com/YunYang1994/tensorflow-yolov3.git
2.在食用代码前,先安装一些依赖项