neural_compressor.experimental.data.datasets.style_transfer_dataset

Dataset used for style transfer task on multiple framework backends.

Module Contents

Classes

StyleTransferDataset

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

class neural_compressor.experimental.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)

Bases: neural_compressor.experimental.data.datasets.dataset.Dataset

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.