基于单幅RGB图像的室内场景三维重建方法及系统

摘要

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

关键词