Sintel¶
- class torchvision.datasets.Sintel(root: Union[str, Path], split: str = 'train', pass_name: str = 'clean', transforms: Optional[Callable] = None)[source]¶
用于光流的 Sintel 数据集。
预计数据集具有以下结构
root Sintel testing clean scene_1 scene_2 ... final scene_1 scene_2 ... training clean scene_1 scene_2 ... final scene_1 scene_2 ... flow scene_1 scene_2 ...
- 参数:
root (str 或
pathlib.Path
) – Sintel 数据集的根目录。split (string, optional) – 数据集分割,可以是“train”(默认)或“test”
pass_name (string, optional) – 要使用的通道,可以是“clean”(默认)、“final”或“both”。有关不同通道的详细信息,请参见上面的链接。
transforms (callable, optional) – 一个函数/转换,它接受
img1, img2, flow, valid_flow_mask
并返回转换后的版本。为了与返回内置有效掩码的其他数据集(如KittiFlow
)保持一致,预计将提供valid_flow_mask
。
- 特殊成员: