空间几何信息估计模型的训练方法及装置

基本信息

申请号 CN202210345349.3 申请日 -
公开(公告)号 CN114758068A 公开(公告)日 2022-07-15
申请公布号 CN114758068A 申请公布日 2022-07-15
分类号 G06T17/00(2006.01)I;G06K9/62(2022.01)I;G06V10/774(2022.01)I 分类 计算;推算;计数;
发明人 谢佳锋;隋伟;陈腾;任伟强;张骞 申请(专利权)人 北京地平线信息技术有限公司
代理机构 北京思源智汇知识产权代理有限公司 代理人 -
地址 100094北京市海淀区丰豪东路9号院2号楼3层1单元301
法律状态 -

摘要

摘要 本公开实施例公开了一种空间几何信息估计模型的训练方法及装置、一种空间几何信息估计方法及装置、计算机可读存储介质、电子设备,其中,该模型的训练方法包括:对样本图像序列中的第一样本图像和第二样本图像进行特征提取,得到特征数据;基于特征数据,由待训练模型进行空间几何信息预测,生成预测空间几何信息;基于预测空间几何信息、预设的标注空间几何信息和预设的有监督的空间几何损失函数,确定空间几何损失值;基于空间几何损失值,调整待训练模型的参数,直至满足预设训练结束条件,得到空间几何信息估计模型。本公开实施例充分利用了多帧图像之间的视差来进行空间几何信息预测,其预测准确性和泛化能力大大提高,预测效率大大提升。