【目标检测·yolo系列】YOLOV2目标检测论文笔记(参考 同济子豪兄的解读)

目录

  • anchor 、 Dimension Clusters 、Direct location prediction
    • 在YOLOv2中规定了五种anchor,
    • grid cell数量为奇数
    • 限制预测框位置
  • 损失函数
  • 细粒度特征
  • Multi-Scale Training

anchor 、 Dimension Clusters 、Direct location prediction

在YOLOv1中先验框是随机的,在YOLOv2中规定了两个先验框,如下图
【目标检测·yolo系列】YOLOV2目标检测论文笔记(参考 同济子豪兄的解读)_第1张图片
如图,①框擅长套高瘦长的物体,②框擅长矮胖宽的物体
由此可以使得模型稳定

在YOLOv2中规定了五种anchor,

并把图片分成了13*13个grid cell,每个grid cell预测5种anchor,每一种anchor对应了一种预测框,每一个预测框只需要输出他相对于其anchor的偏移量。
【目标检测·yolo系列】YOLOV2目标检测论文笔记(参考 同济子豪兄的解读)_第2张图片
如下图

白色汽车的人工标注框(白色框)的中心位置落在了4号框中,那么就由4号框产生的五种anchor中与白框IOU最大的anchor去预测,其预测框只需要输出相比于所属anchor的偏移量即可。
【目标检测·yolo系列】YOLOV2目标检测论文笔记(参考 同济子豪兄的解读)_第3张图片

grid cell数量为奇数

【目标检测·yolo系列】YOLOV2目标检测论文笔记(参考 同济子豪兄的解读)_第4张图片
当图片存在主体的时候,一般是由中心grid cell负责,奇数时正好有中心cell存在,偶数就会落在中心点的位置上——被四个框争取。

【目标检测·yolo系列】YOLOV2目标检测论文笔记(参考 同济子豪兄的解读)_第5张图片
在yolo1中,一个bounding box包含4个维度+一个置信度,另外加上20个预测输出的条件类别概率,共5*2+20=30维

在YOLOv2中,条件类别概率归anchor管,一个anchor中有4个维度+1置信度+20个类别,共5个anchor,总计5*25=125维,一个grid cell有5个anchor共125维

【目标检测·yolo系列】YOLOV2目标检测论文笔记(参考 同济子豪兄的解读)_第6张图片
注:5个anchor通过k-means聚类选出

限制预测框位置

现对于anchor的偏移量tx ty tw th,tx,ty ∈(-∞,∞)
为了防止野蛮生长,就在tx和ty上添加sigmoid函数,使得sigmoid(tx,ty)属于(0,1) 【注】
cx和cy是左上角的坐标,已归一化。由此,预测框的中心位置不会脱离anchor所在的cell。
【目标检测·yolo系列】YOLOV2目标检测论文笔记(参考 同济子豪兄的解读)_第7张图片

置信度:
在这里插入图片描述

【注】:sigmoid函数:
【目标检测·yolo系列】YOLOV2目标检测论文笔记(参考 同济子豪兄的解读)_第8张图片
【目标检测·yolo系列】YOLOV2目标检测论文笔记(参考 同济子豪兄的解读)_第9张图片

损失函数

yolov1:
【目标检测·yolo系列】YOLOV2目标检测论文笔记(参考 同济子豪兄的解读)_第10张图片
yolov2:
【目标检测·yolo系列】YOLOV2目标检测论文笔记(参考 同济子豪兄的解读)_第11张图片

细粒度特征

【目标检测·yolo系列】YOLOV2目标检测论文笔记(参考 同济子豪兄的解读)_第12张图片

【目标检测·yolo系列】YOLOV2目标检测论文笔记(参考 同济子豪兄的解读)_第13张图片
在yolo2的代码中:
【目标检测·yolo系列】YOLOV2目标检测论文笔记(参考 同济子豪兄的解读)_第14张图片

.

Multi-Scale Training

多尺度训练
【目标检测·yolo系列】YOLOV2目标检测论文笔记(参考 同济子豪兄的解读)_第15张图片

每10步重新选择同图不同尺度的图像,强制模型适应不同的大小的图
低分辨率的:快、精度小;
高分辨率的:慢、精度大
【目标检测·yolo系列】YOLOV2目标检测论文笔记(参考 同济子豪兄的解读)_第16张图片

你可能感兴趣的:(深度学习,pytorch,深度学习,YOLO,yolov2,目标检测)