HD1K¶
- class torchvision.datasets.HD1K(root: Union[str, Path], split: str = 'train', transforms: Optional[Callable] = None)[source]¶
用于光流的 HD1K 数据集。
预计数据集具有以下结构
root hd1k hd1k_challenge image_2 hd1k_flow_gt flow_occ hd1k_input image_2
- 参数:
root (str 或
pathlib.Path
) – HD1K 数据集的根目录。split (string, 可选) – 数据集分割,要么是“train”(默认值),要么是“test”。
transforms (callable, 可选) – 一个函数/变换,它接受
img1, img2, flow, valid_flow_mask
并返回一个变换后的版本。
- 特殊成员: