neural_compressor.data.datasets.style_transfer_dataset

Dataset used for style transfer task on multiple framework backends.

Classes

StyleTransferDataset

Dataset used for style transfer task on tensorflow/inteltensorflow/tensorflow_itex backend.

Module Contents

class neural_compressor.data.datasets.style_transfer_dataset.StyleTransferDataset(content_folder, style_folder, crop_ratio=0.1, resize_shape=(256, 256), image_format='jpg', transform=None, filter=None)

Dataset used for style transfer task on tensorflow/inteltensorflow/tensorflow_itex backend.

This Dataset is to construct a dataset from two specific image holders representing

content image folder and style image folder.