基于pytorch实现RGBD数据集的加载(以NYU数据集为例)
NYU数据集NYU数据集由训练集和验证集两部分组成,其中包括了训练图像、深度图像和标签图像三部分组成。其中训练集由795张图像。构建Datasetimport numpy as npimport osfrom torch.utils.data import Datasetfrom PIL import Image# 数据集路径Depth_Train = ' 'Depth_Test = ' 'Image_Train = ' 'Image_Test = ' 'Label_T
原创
2022-03-31 03:40:43 ·
943 阅读 ·
6 评论