《表2 金字塔池化模块的计算量》
接着分析YOLOv2_PPM网络的计算量问题。YOLOv2_PPM网络相对于YOLOv2网络增加了金字塔池化模块的计算。金字塔池化模块的计算量主要来自于卷积层。当卷积层的输入特征图为xi∈?ni×hi×wi,其中ni为通道数,hi和wi分为高和宽。卷积层的通道数为ni+1,卷积核为k×k。则经过该卷积层后的输出特征图为xi+1∈?ni+1×hi+1×wi+1。此时,卷积层的参数为ni+1nik2,卷积层的计算量为ni+1nik2hi+1wi+1。由此,可以得出,卷积层的计算量与输出特征图的维度以及高和宽成正比。由表1得,YOLOv2的计算量为1.432 6×1010次。因为YOLOv2_PPM将YOLOv2的最后一层的参数125×(1 024×1×1)改为1 024×(1 024×1×1),所以增加的计算量为0.015 6×1010次。由于金子塔池化模块降低了特征图的大小,并且其中的卷积层使特征图的维度降低为输入通道数的1/4,因此金字塔池化模块引入的计算量很少,见表2,金字塔池化模块的计算量为0.081 2×1010次。因此YOLOv2_PPM比YOLOv2增加了0.096 8×1010次,增加的运算量为YOLOv2的1/15,对YOLOv2的实时检测速度没有过多的影响。在3.3节中对YOLOv2和YOLOv2_PPM的速度进行对比,后者仅比前者慢了3帧/s,减少的速度为前者的1/14。
图表编号 | XD0035433700 严禁用于非法目的 |
---|---|
绘制时间 | 2019.02.01 |
作者 | 王琳、卫晨、李伟山、张钰良 |
绘制单位 | 西安邮电大学通信与信息工程学院、西安邮电大学经济与管理学院、西安邮电大学通信与信息工程学院、西安邮电大学经济与管理学院 |
更多格式 | 高清、无水印(增值服务) |