国内刊号:51-1196/TP
国际刊号:1001-3695
发布日期:
作者:贾瑞明,李彤,李阳,王一丁,
单位:北方工业大学信息学院,北京100144;
关键词:单目深度估计,卷积神经网络,编解码结构,轻量金字塔解码,
基金:国家自然科学基金面上项目(61673021);北方工业大学学生科技活动资助项目;;
针对单目深度估计网络庞大的参数量和计算量,提出一种轻量金字塔解码结构的单目深度估计网络,可以在保证估计精度的情况下降低网络模型的复杂度、减少运算时间。该网络基于编解码结构,以端到端的方式估计单目图像的深度图。编码端使用ResNet50网络结构;在解码端提出了一种轻量金字塔解码模块,采用深度空洞可分离卷积和分组卷积以提升感受野范围,同时减少了参数量,并且采用金字塔结构融合不同感受野下的特征图以提升解码模块的性能;此外,在解码模块之间增加跳跃连接实现知识共享,以提升网络的估计精度。在NYUD v2数据集上的实验结果表明,与结构注意力引导网络相比,轻量金字塔解码结构的单目深度估计网络在误差RMS的指标上降低约11.0%,计算效率提升约84.6%。
来源:2021年第1期
《计算机应用研究》期刊编辑部