摘要
本发明涉及一种基于单幅RGB图像的室内场景三维重建方法及系统。所述重建方法包括:获取室内场景ScanNet数据集,并对所述ScanNet数据集进行划分,确定训练集以及测试集;利用随机样本一致性RANSAC算法,从所述训练集中重复提取平面,确定室内场景待重建平面;利用局部特征提取分支提取所述室内场景待重建平面的局部特征;所述局部特征提取分支用于提取局部特征;根据所述局部特征以及扩张残差DRN网络确定全局特征;根据所述局部特征以及所述全局特征构建基于单幅RGB图像的室内场景三维重建网络模型;根据所述室内场景三维重建网络模型重建所述单幅RGB图像的室内场景。采用本发明的重建方法及系统能够提高深度图预测精度、图像边缘重建效果以及室内场景重建精度。
- 单位