计算机应用

• 人工智能与仿真 •    下一篇

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

邵伟平1,王兴2,白帆2,曹昭睿2   

  1. 1. 沈阳理工大学机械工程学院
    2. 沈阳理工大学
  • 收稿日期:2019-09-16 修回日期:2019-11-18 发布日期:2019-11-18 出版日期:2020-05-09
  • 通讯作者: 王兴

Design of lightweight convolutional neural network based on MobileNet and YOLOv3

  • Received:2019-09-16 Revised:2019-11-18 Online:2019-11-18 Published:2020-05-09

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

关键词: 深度学习, 卷积神经网络, MobileNet, YOLOv3, 轻量化网络

Abstract: Aiming at the problems of poor compatibility,low computational power and excessive memory occupied in the training process of current convolutional neural network-based target detection algorithms in small image processing computing platform,a ligntweighted convolutional neural network named YOLO-Slim was proposed and its feasibility was verified by YOLOv3. Firstly,the network parameters and computational complexity were greatly reduced by changing the network infrastructure and replacing the standard convolution with the deep separable convolution. Secondly,according to the impact of network layer on mAP(mean Average Precison),the network layer was pruned to achieve inter-layer pruning of network;then the median channel pruning strategy was used to achieve intra-layer pruning of network;finally,the lightweight network design was completed. The experimental results show that the YOLO-Slim designed on VOC2007 test data set is 90% smaller than the original YOLOv3 in model size,76. 42% in mAP and 16 ms in recognition speed. It can provide fast and accurate target recognition capability for small image computing platform.

Key words: deep learning, Convolutional Neural Network (CNN), MobileNet, YOLOv3, lightweight network

中图分类号: