-
PyTorch NYUv2数据集类：pytorch-nyuv2
资源介绍
PyTorch NYUv2数据集类
针对NYUv2数据集的PyTorch包装器专注于多任务学习。 可用的数据源：RGB，语义分割（13），表面法线，深度图像。
从以下位置下载数据：
例子
from nyuv2 import NYUv2
from torchvision import transforms
t = transforms.Compose([transforms.RandomCrop(400), transforms.ToTensor()])
NYUv2(root="/somepath/NYUv2", download=True,
rgb_transform=t, seg_transform=t, sn_transform=t, depth_transform=t)
Dataset NYUv2
Number of datapoints: 795
S