cerebras.modelzoo.data.vision.classification.data.smallnorb.SmallNORB#

class cerebras.modelzoo.data.vision.classification.data.smallnorb.SmallNORB[source]#

Bases: torchvision.datasets.vision.VisionDataset

This database is intended for experiments in 3D object recognition from shape. It contains images of 50 toys belonging to 5 generic categories: four-legged animals, human figures, airplanes, trucks, and cars. The objects were imaged by two cameras under 6 lighting conditions, 9 elevations (30 to 70 degrees every 5 degrees), and 18 azimuths (0 to 340 every 20 degrees).

The training set is composed of 5 instances of each category (instances 4, 6, 7, 8 and 9), and the test set of the remaining 5 instances (instances 0, 1, 2, 3, and 5).

Methods

__init__(root, split='train', task=None, transform=None, target_transform=None)[source]#
__call__(*args: Any, **kwargs: Any) Any#

Call self as a function.

static __new__(cls, *args: Any, **kwargs: Any) Any#