CREStereo¶
- class torchvision.datasets.CREStereo(root: Union[str, Path], transforms: Optional[Callable] = None)[源代码]¶
用于训练 CREStereo 架构的合成数据集。数据集详细信息在官方论文 仓库 中。
预计数据集具有以下结构
root CREStereo tree img1_left.jpg img1_right.jpg img1_left.disp.jpg img1_right.disp.jpg img2_left.jpg img2_right.jpg img2_left.disp.jpg img2_right.disp.jpg ... shapenet img1_left.jpg img1_right.jpg img1_left.disp.jpg img1_right.disp.jpg ... reflective img1_left.jpg img1_right.jpg img1_left.disp.jpg img1_right.disp.jpg ... hole img1_left.jpg img1_right.jpg img1_left.disp.jpg img1_right.disp.jpg ...
- 参数:
root (str) – 数据集的根目录。
transforms (callable, optional) – 一个函数/变换,它接收一个样本并返回一个变换后的版本。
- 特殊成员: