advsecurenet.datasets.ImageNet package


advsecurenet.datasets.ImageNet.imagenet_dataset module

class advsecurenet.datasets.ImageNet.imagenet_dataset.ImageNetDataset(preprocess_config: PreprocessConfig | None = None)

Bases: ImageFolderBaseDataset, BaseDataset

s The ImageNetDataset class that loads the ImageNet dataset.

mean

Mean of the ImageNet dataset.

Type:

List[float]

std

Standard deviation of the ImageNet dataset.

Type:

List[float]

input_size

Input size of the ImageNet images.

Type:

Tuple[int, int]

name

Name of the dataset.

Type:

str

num_classes

Number of classes in the ImageNet dataset.

Type:

int

num_input_channels

Number of input channels in the ImageNet images.

Type:

int