SPP讲解

忘是亡心i 2022-10-05 00:41 15阅读 0赞

### 文章目录 ###

*  yoloV3和yoloV3-spp网络结构对比
 *   *  如何理解yoloV3中的SPP模块？

--------------------

# yoloV3和yoloV3-spp网络结构对比 #

YOLOv3网络结构图：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3pob25nbG9uZ3NoZW4_size_16_color_FFFFFF_t_70]  
YOLOv3-spp网络结构图：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3pob25nbG9uZ3NoZW4_size_16_color_FFFFFF_t_70 1]  
SPP模块结构如下图：  
![在这里插入图片描述][20210607115019358.png]  
从上述两幅网络图中我们可以看出，在相对于普通版本的YOLOV3，SPP版本在第5，6层卷积之间增加了一个SPP模块，这个模块主要是由不同的池化操作组成，具体的实现在yoloV3-SPP的cfg文件中：

### SPP ###
    [maxpool]
    stride=1
    size=5
    
    [route]
    layers=-2
    
    [maxpool]
    stride=1
    size=9
    
    [route]
    layers=-4
    
    [maxpool]
    stride=1
    size=13
    
    [route]
    layers=-1,-3,-5,-6

## 如何理解yoloV3中的SPP模块？ ##

在一般的CNN网络结构中，最后的分类层通常是由全连接层组成，而全连接有个特点，那就是它的特征数是固定的，这就导致了图片在输入网络的时候，大小必须是固定的，但是在实际情况中，图片大小是多种多样的，如果不能满足网络的输入，图片将无法在网络中进行前向运算，所以为了得到固定尺寸的图片，必须对图片进行裁剪或者变形拉伸等，这样就很可能会导致图像失真，从而影响最终的精度，而我们希望网络能够保持图大小的输入，得到最大的精度。  
![在这里插入图片描述][20210607145033995.png]  
SPP全程为空间金字塔池化结构，主要是为了解决两个问题：  
1：有效避免了对图像区域裁剪、缩放操作导致的图像失真等问题；  
2：解决了卷积神经网络对图相关重复特征提取的问题，大大提高了产生候选框的速度，且节省了计算成本。

但是在yoloV3中，并不是解决这两个问题的，在YOLOV3-SPP中，SPP module是由4个并行的分支构成，分别为kernel size为5x5,9x9,13x13的最大池化和一个跳跃链接。如下图所示，作者检测头前面的第5和第6卷积层之间集成SPP模块来获得YOLOv3-SPP，在Feature Map经过SPP module池化后的特征图重新cat起来传到下一层侦测网络中  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3pob25nbG9uZ3NoZW4_size_16_color_FFFFFF_t_70 2]  
所以说，YOLOv3-SPP版本实际上只是增加了SPP 模块，该模块借鉴了空间金字塔的思想，通过SPP模块实现了局部特征和全局特征，这也是为什么SPP模块中最大的池化核大小要尽可能的接近或者等于需要池化的特征图的大小，特征图经过局部特征与全局特征相融合后，丰富了特征图的表达能力，有利于待检测图像中目标大小差异较大的情况，尤其是对于YOLOv3这种复杂的多目标检测，所以对检测的精度上有了很大的提升。

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3pob25nbG9uZ3NoZW4_size_16_color_FFFFFF_t_70]: /images/20221005/249f9111f1c84883aa9dc6d8aad84172.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3pob25nbG9uZ3NoZW4_size_16_color_FFFFFF_t_70 1]: /images/20221005/de69ea84488a489d81cb12b2e79e6964.png
[20210607115019358.png]: /images/20221005/eee43b90982d4b429d8c32b1bf1d975c.png
[20210607145033995.png]: /images/20221005/96ca7887a0da4976a27baad8699f9e17.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3pob25nbG9uZ3NoZW4_size_16_color_FFFFFF_t_70 2]: /images/20221005/4bb163170b0a47b19dcbca3dbee5844e.png