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

被引：45

作者：

邵伟平 ^{[1
]}

王兴 ^{[1
]}

曹昭睿 ^{[2
]}

白帆 ^{[2
]}

机构：

[1] 沈阳理工大学机械工程学院

[2] 沈阳理工大学装备工程学院

来源：

计算机应用 | 2020年 / 40(S1)卷 / S1期

关键词：

深度学习; 卷积神经网络; MobileNet; YOLOv3; 轻量化网络;

D O I：

暂无

中图分类号：

TP183 [人工神经网络与计算];

学科分类号：

摘要：

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

引用

页码：8 / 13

页数：6