目标检测(下)

目标检测的主流算法分为两种类型:

  • two-stage:RCNN系列。通过Selective Search或者RPN产生稀疏的候选框,然后对这些候选框进行分类和回归。
  • one-stage:YOLO和SSD。不做候选区域的选择,而是均匀地在图像不同位置进行密集采样,直接对所有可能的区域进行分类与回归,整个过程只需要一步。

一、SSD特点

1.使用不同尺寸的特征图用于检测
 大的特征图检测小目标,小的特征图用来检测大目标。

2.使用不同尺寸的Anchor框
 每个单元设置尺寸或者长宽比不同的先验框,用来检测不同形状的物体,预测的边界框是以这些先验框为基准的,在一定程度上减小了训练难度。

3.用卷积层进行预测
 SSD网络模型是全卷积网络,没有全连接层,直接使用卷积层做预测。
 1)卷积层做预测:使用图像特征提取卷积层的输出(VGG16去掉最后一层全连接层),预测anchor框包含的物理类别、框的位置及大小
 2)使用不同卷积层的输出预测不同尺寸的物体

4.端到端的训练

二、SSD网络结构

SSD网络模型是全卷积网络:没有全连接层,使用卷积层做预测。

  • 卷积层输出的通道个数 == 每个位置anchor框的个数*(4个位置信息+所有类别个数)
  • 卷积层不同通道输出的值 == 每个位置不同宽高比的预测值(包含4个位置信息+所有类别的概率)
SSD网络结构(1).png
SSD网络结构(2).PNG

三、SSD网络结构

1.SSD网络结构详解

SSD网络结构.png
  • SSD是一个没有全连接层的网络
  • 在VGG-16网络模型的基础上,将VGG-16的第一个卷积层FC6和第二个卷积层FC7,替换为卷积层conv6和conv7
  • 去掉最后一个用于分类的全连接层FC8
  • 加上了4组卷积层:conv8、conv9、conv10、conv11

2.SSD卷积层做预测

 普通神经网络结构是在最后一层使用全连接层做预测,而SSD网络是使用多个卷积网络直接做预测。

SSD预测.PNG

四、SSD网络原理

1.卷积如何做预测?

 通过设置卷积核的个数(即通道数),得到预测值的信息。每个卷积核都代表一个待预测的信息,比如位置信息x、y或者属于某一个类别的概率值p。

  • 预测目标所属类别
卷积层做预测.PNG
  • 预测目标边界框
卷积层预测多个值.PNG
预测值的数量为:
  • 每一个位置需要预测(4+类别总数)个值
  • 每个位置又有k个anchor框,需要预测k*(4+类别总数)个值
  • 每个特征图又有m*n个像素,所以需要预测m*n*k*(4+类别总数)个值

五、SSD损失函数

SSD损失函数.png

N:与真实边界框配对的anchor的数量
x:如果一个anchor与真实边界框配对,值为1,否则为0
c:真实物体的预测值
l:预测的边界框中心位置和长、宽
g:真实边界框中心位置和长、宽

1.定位损失函数

  • 边框位置的预测使用相对值
    因为绝对值的范围比较大,边框的长宽可能是几十个像素,而神经网络的输入需要归一化后的值,效果才会好,所以边框的预测是使用(0~1)之间的相对值。
边框预测.png
  • Smooth L1损失函数
    结合了L1的优势:随着x的值增大,梯度的增长是恒定的
    结合了L2的优势:当x特别小的时候,梯度迅速减小,减少震荡
Smooth损失函数.png

2.分类损失函数

分类损失函数.png

六、anchor的框选择

1.anchor大小计算公式

 假设最后选择m个特征图做预测,那么每一个特征图对应的anchor大小按照如下公式进行选择:
\begin{align*}\label{2} & s_{k} = s_{min} + \frac{s_{max}-s_{min}}{m-1}\left ( k-1 \right ) \\ &\\ & s_{min} = 0.1 \\ & s_{max} = 0.95 & \end{align*}

  • anchor大小计算案例:
anchor大小计算.PNG

备注:
○ k=1时,anchor尺寸 = 原图*0.1
○ k=m时,anchor尺寸 = 原图*0.95
○ 即,anchor框的尺寸与特征图的大小有关,越靠后的特征图尺寸越小,anchor框占特征图的比例越大。这也就解释了,为什么前面大的特征图适合检测小物体,后面小的特征图适合检测大物体。

2. anchor框比例选择

anchor框的比例.png

3.anchor中心位置计算公式

anchor的中心位置.PNG

七、生成负样本

生成负样本.png

八、空洞卷积

空洞卷积.png
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 217,657评论 6 505
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,889评论 3 394
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 164,057评论 0 354
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,509评论 1 293
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,562评论 6 392
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,443评论 1 302
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,251评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,129评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,561评论 1 314
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,779评论 3 335
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,902评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,621评论 5 345
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,220评论 3 328
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,838评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,971评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,025评论 2 370
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,843评论 2 354

推荐阅读更多精彩内容