[−][src]Module tch::vision::dataset
A simple dataset structure shared by various computer vision datasets.
Structs
| Dataset |
Functions
| augmentation | |
| random_crop | Pad the image using reflections and take some random crops. This expects a 4 dimension NCHW tensor and returns a tensor with an identical shape. |
| random_cutout | Applies cutout: randomly remove some square areas in the original images. https://arxiv.org/abs/1708.04552 |
| random_flip | Randomly applies horizontal flips This expects a 4 dimension NCHW tensor and returns a tensor with an identical shape. |