FGVCAircraft

class torchvision.datasets.FGVCAircraft(root: str, split: str = 'trainval', annotation_level: str = 'variant', transform: Optional[Callable] = None, target_transform: Optional[Callable] = None, download: bool = False)

FGVC Aircraft Dataset.

The dataset contains 10,200 images of aircraft, with 100 images for each of 102 different aircraft model variants, most of which are airplanes. Aircraft models are organized in a three-levels hierarchy. The three levels, from finer to coarser, are:

• variant, e.g. Boeing 737-700. A variant collapses all the models that are visually

  indistinguishable into one class. The dataset comprises 102 different variants.

• family, e.g. Boeing 737. The dataset comprises 70 different families.

• manufacturer, e.g. Boeing. The dataset comprises 41 different manufacturers.

Parameters

• root (string) – Root directory of the FGVC Aircraft dataset.

• split (string, optional) – The dataset split, supports train, val, trainval and test.

• annotation_level (str, optional) – The annotation level, supports variant, family and manufacturer.

• transform (callable, optional) – A function/transform that takes in an PIL image and returns a transformed version. E.g, transforms.RandomCrop

• target_transform (callable, optional) – A function/transform that takes in the target and transforms it.

• download (bool, optional) – If True, downloads the dataset from the internet and puts it in root directory. If dataset is already downloaded, it is not downloaded again.

Special-members

__getitem__(idx) → Tuple[Any, Any]

Parameters

index (int) – Index

Returns

Sample and meta data, optionally transformed by the respective transforms.

Return type

(Any)