目标检测算法 - YOLOv3

121 阅读1分钟

目标检测算法 - YOLOv3

YOLOv1、YOLOv2都是在CVPR这种正规的计算机视觉学术会议上发表的正式学术论文。

YOLOv3不算一篇严谨的学术论文,是作者随笔写的技术报告。

YOLOv3性能:


1. Backbone Darknet-53

YOLOv3在v2的基础上,更换了骨干网络,将Darknet-19替换为了Darknet-53。

|604

性能对比:

|550


2. 整体架构

输入为416*416大小的图像。输出3个尺度的feature map,分别为13×13×25526×26×25552×52×25513\times 13\times 255,26\times 26\times 255,52\times 52\times 255

三个尺度,分别输入图像划分为13×1326×2652×5213\times 13,26\times 26,52\times 52个grid cell。

13×1313\times 13下采样32倍,26×2626\times 26下采样16倍,52×5252\times 52下采样8倍。

13×1313\times 13的感受野对应原图上的感受野就是32×3232\times 32

13×1313\times13负责预测大物体,26×2626\times26负责预测中等大小物体,52×5252\times52负责预测小物体。

255怎么来的呢?

3×(5+80)=2553\times (5+80)=255。其中3:每个grid cell生成3个anchor,每个anchor对应一个预测框,每个预测框有5+80个参数,5:(x,y,w,h,c),80:coco数据集80个类别的条件类别概率 。


3. 损失函数

在yolov3中,如果一个anchor和ground-truth的IoU最大,那么它就是正样本。如果它和ground-truth有一

部分IoU,但是不是最大的,那么这些框则忽略掉。如果某个anchor和ground-truth的IoU小于阀值,它则为负样本。

|514


4. 训练过程


5. 预测过程