neural_compressor.data.datasets.imagenet_dataset

Dataset for ImageNet data generation on multiple framework backends.

Classes

ImagenetRaw	Configuration for ImageNet raw dataset.
PytorchImagenetRaw	Dataset for ImageNet data generation on pytorch backend.
MXNetImagenetRaw	Dataset for ImageNet data generation on mxnet backend.
TensorflowImagenetRaw	Dataset for ImageNet data generation on tensorflow/inteltensorflow/tensorflow_itex backend.
TensorflowImagenetDataset	Configuration for Imagenet dataset.
ONNXRTImagenetDataset	Configuration for Imagenet dataset.

Module Contents

class neural_compressor.data.datasets.imagenet_dataset.ImagenetRaw(data_path, image_list, transform=None, filter=None)

Configuration for ImageNet raw dataset.

Please arrange data in this way:

data_path/img1.jpg data_path/img2.jpg … data_path/imgx.jpg

dataset will read name and label of each image from image_list file, if user set image_list to None, it will read from data_path/val_map.txt automatically.

class neural_compressor.data.datasets.imagenet_dataset.PytorchImagenetRaw(data_path, image_list, transform=None, filter=None)

Dataset for ImageNet data generation on pytorch backend.

class neural_compressor.data.datasets.imagenet_dataset.MXNetImagenetRaw(data_path, image_list, transform=None, filter=None)

Dataset for ImageNet data generation on mxnet backend.

class neural_compressor.data.datasets.imagenet_dataset.TensorflowImagenetRaw(data_path, image_list, transform=None, filter=None)

Dataset for ImageNet data generation on tensorflow/inteltensorflow/tensorflow_itex backend.

class neural_compressor.data.datasets.imagenet_dataset.TensorflowImagenetDataset

Configuration for Imagenet dataset.

class neural_compressor.data.datasets.imagenet_dataset.ONNXRTImagenetDataset(root, subset='val', num_cores=28, transform=None, filter=None)

Configuration for Imagenet dataset.