SceneFlowStereo¶
- class torchvision.datasets.SceneFlowStereo(root: Union[str, Path], variant: str = 'FlyingThings3D', pass_name: str = 'clean', transforms: Optional[Callable] = None)[源代码]¶
用于 Scene Flow 数据集的数据集接口。此接口提供对 FlyingThings3D、`Monkaa 和 Driving 数据集的访问。
数据集预计具有以下结构
root SceneFlow Monkaa frames_cleanpass scene1 left img1.png img2.png right img1.png img2.png scene2 left img1.png img2.png right img1.png img2.png frames_finalpass scene1 left img1.png img2.png right img1.png img2.png ... ... disparity scene1 left img1.pfm img2.pfm right img1.pfm img2.pfm FlyingThings3D ... ...
- 参数:
root (str 或
pathlib.Path
) – SceneFlow 所在的根目录。variant (string) – 要使用的哪个数据集变体,“FlyingThings3D”(默认)、“Monkaa” 或“Driving”。
pass_name (string) – 要使用的哪个通道, “clean”(默认)、“final” 或“both”。
transforms (callable, optional) – 一个函数/变换,它接受一个样本并返回一个变换后的版本。
- 特殊成员: