摘要
本申请提供了一种深度优化模型的训练方法及深度估计图像的优化方法,属于人工智能技术领域。所述方法包括:获取样本单目RGB图像及其对应的样本深度估计图像及样本抠图图像,样本单目RGB图像具有深度优化标签图像;将样本单目RGB图像、样本深度估计图像及样本抠图图像输入到深度优化模型中,输出主体深度预测图像和深度优化预测图像;基于主体深度预测图像、样本抠图图像、深度优化标签图像、深度优化预测图像及总目标损失函数,对深度优化模型进行训练,得到训练好的深度优化模型。本申请提供了一种深度优化模型的训练方法,采用该方法所训练的深度优化模型所优化的深度优化图像的精度较高。
技术关键词
图像
样本
子模块
上采样
像素点
编码模块
输入输出模块
标签
计算机程序产品
通道
阶段
可读存储介质
人工智能技术
尺寸
处理器
优化装置
训练装置
存储器