spyrit.misc.statistics.stat_imagenet

spyrit.misc.statistics.stat_imagenet(stat_root=PosixPath('stats'), data_root=PosixPath('data/ILSVRC2012_img_test_v10102019'), img_size: int = 64, batch_size: int = 1024, get_size: str = 'resize', n_loop: int = 1, device=device(type='cpu'))[source]

Args:

stat_root: path to the folder where the mean and covariance matrices are saved

data_root: path to image database. data_root needs to have all images in a subfolder

img_size: image size

batch_size: batch size

get_size: specifies how images of size img_size are obtained (see data_loaders_imagenet)

Example:

>>> data_root =  Path('../data/ILSVRC2012_img_test_v10102019/')
>>> stat_root =  Path('../stat/ILSVRC2012_img_test_v10102019')
>>> from spyrit.misc.statistics import stat_imagenet
>>> stat_imagenet(stat_root = stat_root, data_root = data_root)