文件名称：pytorch-nyuv2:PyTorch NYUv2数据集类
文件大小：6KB
文件格式：ZIP
更新时间：2024-05-24 12:59:04
pytorch dataset nyuv2 Python
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
【文件预览】：
pytorch-nyuv2-master
----.gitignore(35B)
----requirements.txt(201B)
----LICENSE(1KB)
----README.md(2KB)
----nyuv2()
--------__init__.py(30B)
--------nyuv2.py(11KB)