weixin_44422920的博客

NYU数据集 NYU数据集由训练集和验证集两部分组成，其中包括了训练图像、深度图像和标签图像三部分组成。其中训练集由795张图像。 构建Dataset import numpy as np import os from torch.utils.data import Dataset from PIL import Image # 数据集路径 Depth_Train = ' ' Depth_Test = ' ' Image_Train = ' ' Image_Test = ' ' Label_T