[−][src]Module tch::vision
The vision
module groups functions and models related to
computer vision.
Modules
alexnet | AlexNet. https://arxiv.org/abs/1404.5997 |
cifar | The CIFAR-10 dataset. |
dataset | A simple dataset structure shared by various computer vision datasets. |
densenet | DenseNet implementation. |
efficientnet | EfficientNet implementation. |
image | Utility functions to manipulate images. |
imagenet | Helper functions for ImageNet like datasets. |
inception | InceptionV3. |
mnist | The MNIST hand-written digit dataset. |
mobilenet | MobileNet V2 implementation. https://ai.googleblog.com/2018/04/mobilenetv2-next-generation-of-on.html |
resnet | ResNet implementation. |
squeezenet | SqueezeNet implementation. |
vgg | VGG models. |