发明公开
CN112967327A 基于联合自注意力机制的单目深度方法
无效 - 驳回
- 专利标题: 基于联合自注意力机制的单目深度方法
-
申请号: CN202110239390.8申请日: 2021-03-04
-
公开(公告)号: CN112967327A公开(公告)日: 2021-06-15
- 发明人: 张玉亮 , 赵智龙 , 付炜平 , 孟荣 , 范晓丹 , 刘洪吉 , 张宁 , 王东辉 , 张东坡 , 李兴文 , 曾建生
- 申请人: 国网河北省电力有限公司检修分公司 , 国家电网有限公司
- 申请人地址: 河北省石家庄市新华区钟盛路66号;
- 专利权人: 国网河北省电力有限公司检修分公司,国家电网有限公司
- 当前专利权人: 国网河北省电力有限公司检修分公司,国家电网有限公司
- 当前专利权人地址: 河北省石家庄市新华区钟盛路66号;
- 代理机构: 石家庄新世纪专利商标事务所有限公司
- 代理商 张晓龙
- 主分类号: G06T7/50
- IPC分类号: G06T7/50 ; G06N3/04 ; G06N3/08
摘要:
本发明涉及一种基于联合自注意力机制的单目深度估计方法,其包括:基于联合注意力模块的编码器、基于U‑net的解码器和一个基于特征金字塔的连接模块组成;按所需深度图的分辨率设置各卷积层参数;搭建估计网络模型;使用训练数据集训练网络模型,提取解码器的输出;计算解码器输出和其对应深度图,结合损失函数对模型参数进行修改;使用训练完成后的最终模型对输入图像进行深度预测。本发明在编码器模块中使用了空间自注意力机制、通道自注意力机制并引入过滤机制从而利用局部特征映射并结合全局上下信息来提取深度信息,解决了在卷积神经网络中无法有效集成局部信息和全局信息的问题,进一步提高深度估计的精度。