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