激光与光电子学进展, 2019, 56 (9): 091003, 网络出版: 2019-07-05
基于卷积神经网络的低参数量实时图像分割算法 下载: 1195次
Low-Parameter Real-Time Image Segmentation Algorithm Based on Convolutional Neural Network
图像处理 图像分割 实时图像 低参数量 卷积模块 多尺度特征 image processing image segmentation real-time image low number of parameters convolution module multiscale feature
摘要
提出了一种低参数量实时图像语义分割网络模型Atrous-squeezeseg。模型在最低参数量为2.1×10
7时的运算帧率为45.3 frame/s,像素点准确度与均交并比分别可达到59.5%与62.9%。同时,嵌入式设备NVIDIA TX2的运算帧率可达8.3 frame/s。实验结果表明,相比于其他分割算法,所提模型的速度和参数量均得到了提升。
Abstract
We propose a real-time image semantic segmentation network model, which is named as Atrous-squeezeseg. Under the condition that the minimum parameter of the model is 2.1×10
7, the operation frame rate is 45.3 frame/s, and the pixel point accuracy and mean intersection over union can reach 59.5% and 62.9%, respectively. At the same time, in the embedded device NVIDIA TX2, the operate frame rate is up to 8.3 frame/s. The experimental results show that, compared with other segmentation algorithms, the speed and parameter quantity of the proposed model are increased.
谭光鸿, 侯进, 韩雁鹏, 罗朔. 基于卷积神经网络的低参数量实时图像分割算法[J]. 激光与光电子学进展, 2019, 56(9): 091003. Guanghong Tan, Jin Hou, Yanpeng Han, Shuo Luo. Low-Parameter Real-Time Image Segmentation Algorithm Based on Convolutional Neural Network[J]. Laser & Optoelectronics Progress, 2019, 56(9): 091003.