光学学报, 2019, 39 (4): 0415006, 网络出版: 2019-05-10
面向嵌入式平台的轻量级目标检测网络 下载: 1226次
Light-Weight Object Detection Networks for Embedded Platform
机器视觉 目标检测 深度神经网络 嵌入式系统 实时性 machine vision object detection deep neural network embedded system real time
摘要
基于深度可分离卷积,提出了一种适用于嵌入式平台的小型目标检测网络MTYOLO(MobileNet Tiny-Yolo),它将待检测的图片平均分割成多个单元格,并采用深度可分离卷积代替传统卷积,减少了参数量和计算量。采用点卷积和特征图融合的方法来提高检测精度。实验结果表明,所提MTYOLO网络模型大小为41 MB,约为Tiny-Yolo模型的67%,其在PASCAL VOC 2007数据集上的检测准确率可达到57.25%,检测效果优于Tiny-Yolo模型,更适合应用于嵌入式系统。
Abstract
Based on depth separable convolution, a small object detection network for embedded platform, MTYOLO (MobileNet Tiny-Yolo), is proposed. It divides the image into many grids and replaces the traditional convolution by the depth separable convolution, which decreases the number of parameters and computational cost. The point convolution and the feature map merging are adopted to improve the detection accuracy. The experimental results show that the size of the proposed MTYOLO network model is 41 MB, approximately 67% of that of Tiny-Yolo model. Furthermore, its detection accuracy on the PASCAL VOC 2007 dataset is up to 57.25%, superior to the Tiny-Yolo model’s. The proposed model is particularly suitable for application in embedded platforms.
崔家华, 张云洲, 王争, 刘及惟. 面向嵌入式平台的轻量级目标检测网络[J]. 光学学报, 2019, 39(4): 0415006. Jiahua Cui, Yunzhou Zhang, Zheng Wang, Jiwei Liu. Light-Weight Object Detection Networks for Embedded Platform[J]. Acta Optica Sinica, 2019, 39(4): 0415006.