基于MobileNet与YOLOv3的轻量化卷积神经网络设计

• 作 者: 邵伟平 王兴 曹昭睿 白帆
• 机 构: 沈阳理工大学机械工程学院,沈阳理工大学装备工程学院
• 出 处: 计算机应用
• 发表日期: 2020年S1期
• 关键词: 深度学习 卷积神经网络 MobileNet YOLOv3 轻量化网络 deep learningConvolutional Neural Network(CNN
• 基金项目: 装备预研兵器工业联合基金资助项目(614B012841)

针对当前基于卷积神经网络的目标检测算法在小型图像处理计算平台中兼容性较差、计算能力低下以及网络训练过程中占用内存过大的问题,提出了一种轻量化卷积神经网络(CNN)YOLO-Slim,并利用YOLOv3验证可行性。首先,通过网络基础构架的改变以及将标准卷积替换为深度可分离卷积实现了网络参数与计算量的大幅度降低;其次,依据网络层对平均精度均值(mAP)的影响程度剪枝网络层,实现网络的层间剪枝;然后,使用中位数的通道剪枝策略实现对网络的层内剪枝,最终,完成轻量化网络的设计。实验结果表明,在VOC2007测