InStereo2k¶
- class torchvision.datasets.InStereo2k(root: Union[str, Path], split: str = 'train', transforms: Optional[Callable] = None)[source]¶
InStereo2k 数据集。
预计数据集具有以下结构
root InStereo2k train scene1 left.png right.png left_disp.png right_disp.png ... scene2 ... test scene1 left.png right.png left_disp.png right_disp.png ... scene2 ...
- 参数:
root (
str
或pathlib.Path
) – InStereo2k 所在的根目录。split (string) – “train” 或 “test”。
transforms (callable, optional) – 一个函数/变换,它接收一个样本并返回一个变换后的版本。
- 特殊成员: