opendataval.dataloader.datasets.imagesets#

TorchVision data sets.

Uses torchvision. as a dependency.

Module Attributes

numbers

Vision Classification data set registered as "mnist", from TorchVision.

fashion

Vision Classification data set registered as "fashion", from TorchVision.

cifar100

Vision Classification data set registered as "cifar100", from TorchVision.

cifar10

Vision Classification registered as "cifar10", from TorchVision.

cifar10_embed(cache_dir, force_download, ...)

Vision Classification registered as "cifar10-embeddings" ResNet50 embeddings

stl10_embed(cache_dir, force_download, ...)

Vision Classification registered as "stl10-embeddings" ResNet50 embeddings

svhn_embed(cache_dir, force_download, *args, ...)

Vision Classification registered as "svhn-embeddings" ResNet50 embeddings

Functions

ResnetEmbeding(dataset_class[, size, batch_size])

Convert PIL color Images into embeddings with ResNet50 model.

cifar10_embed(cache_dir, force_download, ...)

Vision Classification registered as "cifar10-embeddings" ResNet50 embeddings

show_image(imgs)

Displays an image or a list of images.

stl10_embed(cache_dir, force_download, ...)

Vision Classification registered as "stl10-embeddings" ResNet50 embeddings

svhn_embed(cache_dir, force_download, *args, ...)

Vision Classification registered as "svhn-embeddings" ResNet50 embeddings

Classes

VisionAdapter(dataset_class)

Adapter for PyTorch vision data sets.