neodroidvision.data.classification.imagenet.imagenet_2012.ImageNet2012

class neodroidvision.data.classification.imagenet.imagenet_2012.ImageNet2012(dataset_path: Path, split: SplitEnum = SplitEnum.training, resize_s: int = 256, crop_size: int = 224)[source]

Bases: SupervisedDataset

description

__init__(dataset_path: Path, split: SplitEnum = SplitEnum.training, resize_s: int = 256, crop_size: int = 224)[source]
Parameters

  • dataset_path – dataset directory

  • split – train, valid, test

Methods

__init__(dataset_path[, split, resize_s, ...])

type resize_s

int or tuple(w,h)

Attributes

base_transform

category_id

category_names

inverse_base_transform

mean

predictor_shape

return: :rtype:

response_shape

return: :rtype:

split_names

return: :rtype:

std

property predictor_shape: Tuple[int, ...]

return: :rtype:

property response_shape: Tuple[int, ...]

return: :rtype:

property split_names: Dict[SplitEnum, str]

return: :rtype: