摘要

针对目标检测算法SSD在交通应用中检测精度不高、对小尺度汽车和行人检测能力不足的问题,提出一种改进的SSD检测算法,将原SSD基础网络VGG-16替换成残差网络ResNet-50,来提高特征提取网络提取特征的能力并防止网络衰退。算法额外设计5层卷积层来简化原SSD网络结构,进行多尺度特征图的检测;将注意力机制CBAM融合到新基础网络中以提高语义信息提取能力、小目标检测能力及整体网络精度。实验结果表明,改进算法在KITTI数据集上的检测精度达到62.0%,比原来SSD网络提高4.7%。