Module tch::vision

source ·
Expand description

The vision module groups functions and models related to computer vision.

Modules

AlexNet. https://arxiv.org/abs/1404.5997
The CIFAR-10 dataset.
ConvMixer implementation.
A simple dataset structure shared by various computer vision datasets.
DenseNet implementation.
EfficientNet implementation.
Utility functions to manipulate images.
Helper functions for ImageNet like datasets.
InceptionV3.
The MNIST hand-written digit dataset.
MobileNet V2 implementation. https://ai.googleblog.com/2018/04/mobilenetv2-next-generation-of-on.html
ResNet implementation.
SqueezeNet implementation.
VGG models.