技术领域
本发明属于智能体自主导航和环境感知领域,具体涉及一种基于CBAM的单目无监督深 度估计方法。
背景技术
智能体为实现安全可靠的自主导航需要具备完善的环境感知功能,这其中就包括对智能 体周围的环境进行深度估计。基于三维激光雷达的深度估计能够得到较为精确的深度估计结 果,但是其价格昂贵且只能得到稀疏的深度估计。基于RGBD相机的深度估计操作简单,但 深度估计范围有限且在室外环境中使用受限。基于立体相机的深度估计在室内外使用不受限 制,但其对计算资源的占用较大,且由于基线偏短造成深度估计范围有限。基于单目相机的 深度估计能够得到稠密的深度图,但是由于缺乏绝对尺度,传统的单目方法不能得到真实的 深度估计。
随着人工智能的发展,智能体逐渐应用深度卷积神经网络来完成环境感知任务。研究人 员最早利用监督学习来恢复单目相机的绝对尺度,从而完成单目的稠密深度估计。但是监督 学习需要大量具有GroundTruth的数据样本进行训练,这大大制约了它的泛化能力。目前, 基于无监督的单目深度估计以其简单有效的训练方式和渐趋提高的精度表现得到了研究者们 的青睐,各类先进的网络设计思想被运用至其中,比如注意力机制、多路径连接和空间搜索 等。因此研究带注意力机制的单目无监督深度估计方法,以实现智能体对周围环境的高精度 稠密深度感知,具有重要的科研价值和现实意义。
发明内容
为解决上述问题,本发明公开了一种基于CBAM(Convolutional Block AttentionModule, 卷积块的注意力模块)的单目无监督深度估计方法,将注意力机制引入深度估计任务,保留 物体的深度细节和提高深度估计的整体精度,为智能体自主导航和环境感知提供基础。
为达到上述目的,本发明的技术方案如下:
一种基于CBAM的单目无监督深度估计方法,包括以下步骤:
步骤1),引入CBAM同Resblock组合成Resblock-CBAM;
步骤2),基于Resblock-CBAM设计带注意力机制的深度估计网络;
步骤3),针对立体图像对的光度重建、视差平滑和左右视差一致性对深度估计网络进行 训练,并完成对单目图像的深度估计。
进一步的,步骤1)所述的引入CBAM同Resblock组合成Resblock-CBAM,包括如下具体步骤:
a),设置CBAM中的通道注意子模块和空间注意子模块为顺序连接,然后将CBAM与Resblock并行连接形成Conventional Resblock-CBAM,最后的输出方程如式(1)所示:
F
式中,
b),设置CBAM中的通道注意子模块和空间注意子模块为顺序连接,然后将CBAM与Resblock串行连接形成Modified Resblock-CBAM,最后的输出方程如式(2)所示:
F
式中,
c),CBAM中通道注意子模块和空间注意子模块的具体过程如式(3)所示:
式中
通道注意子模块的具体过程如式(4)所示:
式中σ表示sigmoid函数,MLP为多成感知器,ω
空间注意子模块的具体过程如式(5)所示:
式中f
步骤2),基于Resblock-CBAM设计带注意力机制的深度估计网络,包括如下具体步骤:
a),在深度估计网络的编码器中顺序使用四个Resblock-CBAM,前三个是Conventional Resblock-CBAM,第四个是Modified Resblock-CBAM;
b),在深度估计网络的编码器中使用五个跳接层,其中第一个跳接层连接编码器中第一 个卷积层和解码器中第二个上卷积层,第二个跳接层连接编码器中第一个池化层和解码器中 第三个上卷积层,第三个跳接层连接第一个Conventional Resblock-CBAM和解码器中第四个 上卷积层,第四个跳接层连接第二个Conventional Resblock-CBAM和解码器中第五个上卷积 层,第五个跳接层连接第三个Conventional Resblock-CBAM和解码器中第六个上卷积层,而 Modified Resblock-CBAM不含跳接层直接与解码器相连接。
步骤3),针对立体图像对的光度重建、视差平滑和左右视差一致性对深度估计网络进行 训练,并在测试时对单目图像深度估计,包括如下具体步骤:
a),深度估计网络的总训练损失包括光度重建损失、视差平滑损失和左右视差一致性损 失,如式(6)所示:
式中L为深度估计网络的总训练损失,L
b),使用图像光度重建损失衡量输入源图像和其相对应的重建图像之间的差异,左图像 的光度重建损失如式(7)所示:
式中
c),使用视差平滑损失改善深度图在图像梯度处的陡变和不连续,左图像的视差平滑损 失如式(8)所示:
式中
d),使用左右视差一致性损失提高网络对深度图的估计精度,左图像的左右视差一致性 损失如式(9)所示:
式中
本发明的有益效果是:
本发明所述的一种基于CBAM的单目无监督深度估计方法,将卷积块注意力模块CBAM 引入无监督深度估计框架,实现对单目图像的稠密深度估计。在引入CBAM至深度估计网络 的过程中,将CBAM同Resblock组合成Resblock-CBAM,从空间和通道两个维度对输入进 行特征提取;同时,采用跳接对多尺度信息进行融合。使用本发明所提出的方法,将注意力 机制融入深度估计网络并进行基于图像对的光度重建、视差平滑和左右视差一致性等无监督 训练,能够保留环境中物体的深度细节和提高整体的深度估计精度。
附图说明
图1无监督深度估计框架图;
图2残差块与卷积块注意力模块的结合示意图;
图3卷积块注意力模块中子模块示意图;
图4深度估计网络结构图;
图5深度估计可视质量评测图;
图6深度估计实验平台图;
图7真实城市场景深度估计可视图;
图8深度估计方法精度对比表。
具体实施方式
下面结合附图和具体实施方式,进一步阐明本发明,应理解下述具体实施方式仅用于说 明本发明而不用于限制本发明的范围。
本发明所述的一种基于CBAM的单目无监督深度估计方法,如图1所示,首先引入CBAM 同Resblock组合成Resblock-CBAM,然后基于Resblock-CBAM设计带注意力机制的深度估 计网络,最后针对立体图像对的光度重建、视差平滑和左右视差一致性对深度估计网络进行 训练,并完成对单目图像的深度估计;包括如下具体步骤:
步骤1),引入CBAM同Resblock组合成Resblock-CBAM,包括如下具体步骤:
a),设置CBAM中的通道注意子模块和空间注意子模块为顺序连接,然后将CBAM与Resblock并行连接形成Conventional Resblock-CBAM,最后的输出方程如式(1)所示:
式中,
b),设置CBAM中的通道注意子模块和空间注意子模块为顺序连接,然后将CBAM与Resblock串行连接形成Modified Resblock-CBAM,最后的输出方程如式(2)所示:
式中,
c),CBAM中通道注意子模块和空间注意子模块的具体过程如式(3)所示:
式中
通道注意子模块的具体过程如式(4)所示:
式中σ表示sigmoid函数,MLP为多成感知器,ω
空间注意子模块的具体过程如式(5)所示:
式中f
步骤2),基于Resblock-CBAM设计带注意力机制的深度估计网络,包括如下具体步骤:
a),在深度估计网络的编码器中顺序使用四个Resblock-CBAM,前三个是Conventional Resblock-CBAM,第四个是Modified Resblock-CBAM;
b),在深度估计网络的编码器中使用五个跳接层,其中第一个跳接层连接编码器中第一 个卷积层和解码器中第二个上卷积层,第二个跳接层连接编码器中第一个池化层和解码器中 第三个上卷积层,第三个跳接层连接第一个Conventional Resblock-CBAM和解码器中第四个 上卷积层,第四个跳接层连接第二个Conventional Resblock-CBAM和解码器中第五个上卷积 层,第五个跳接层连接第三个Conventional Resblock-CBAM和解码器中第六个上卷积层,而 Modified Resblock-CBAM不含跳接层直接与解码器相连接。
步骤3),针对立体图像对的光度重建、视差平滑和左右视差一致性对深度估计网络进行 训练,并在测试时对单目图像深度估计,包括如下具体步骤:
a),深度估计网络的总训练损失包括光度重建损失、视差平滑损失和左右视差一致性损 失,如式(6)所示:
式中L为深度估计网络的总训练损失,L
b),使用图像光度重建损失衡量输入源图像和其相对应的重建图像之间的差异,左图像 的光度重建损失如式(7)所示:
式中
c),使用视差平滑损失改善深度图在图像梯度处的陡变和不连续,左图像的视差平滑损 失如式(8)所示:
式中
d),使用左右视差一致性损失提高网络对深度图的估计精度,左图像的左右视差一致性 损失如式(9)所示:
式中
本实施例中无监督深度估计框架运行于TensorFlow,选用11GB内存的NVIDIAGeForce RTX 2080Ti型显卡训练网络约22小时完成收敛。设定训练的损失函数中相关权重参数如下: α
Abs Rel:
RMSE Log:
选用含有29个场景共697张图片的Eigen split测试集对在KITTI数据集上训练好的网络 进行测试,并同其它已存在的几类方法进行精度对比和可视质量评测。图8是所提方法与其 它深度估计方法在Eigen split测试集上的精度对比,表中a1,a2,a3各表示δ<1.25,δ<1.25
为了更好地展示无监督深度估计优于监督深度估计的跨场景泛化能力,将在KITTI数据 集上训练好的网络在南京部分城市道路场景下进行深度估计实验。图6是进行真实环境下深 度估计的实验平台,所提方法的深度估计结果如图7所示,已训练好的网络在未知场景下深 度估计的可视质量令人满意,能够保留多数近处物体的深度细节。
本发明方案所公开的技术手段不仅限于上述实施方式所公开的技术手段,还包括基于以 上技术特征做出的改进和润饰,这些改进和润饰也应视为本发明的保护范围。
机译: 基于语境关注机制的单眼无监督深度估计方法
机译: 基于深度学习的使用自我监督学习的道路面积估计装置和方法
机译: 基于自适应集成监督学习的深度神经网络的siso-ofdm Siso-ofdm信道估计装置及其方法