vision.rst

Vision DataModules

The following are pre-built datamodules for computer-vision.

---

Supervised learning

These are standard vision datasets with the train, test, val splits pre-generated in DataLoaders with the standard transforms (and Normalization) values

BinaryEMNIST

pl_bolts.datamodules.binary_emnist_datamodule.BinaryEMNISTDataModule

BinaryMNIST

pl_bolts.datamodules.binary_mnist_datamodule.BinaryMNISTDataModule

CityScapes

pl_bolts.datamodules.cityscapes_datamodule.CityscapesDataModule

CIFAR-10

pl_bolts.datamodules.cifar10_datamodule.CIFAR10DataModule

EMNIST

pl_bolts.datamodules.emnist_datamodule.EMNISTDataModule

FashionMNIST

pl_bolts.datamodules.fashion_mnist_datamodule.FashionMNISTDataModule

Imagenet

pl_bolts.datamodules.imagenet_datamodule.ImagenetDataModule

MNIST

pl_bolts.datamodules.mnist_datamodule.MNISTDataModule

Semi-supervised learning

The following datasets have support for unlabeled training and semi-supervised learning where only a few examples are labeled.

Imagenet (ssl)

pl_bolts.datamodules.ssl_imagenet_datamodule.SSLImagenetDataModule

STL-10

pl_bolts.datamodules.stl10_datamodule.STL10DataModule