neural_compressor.data.datasets.style_transfer_dataset¶
Dataset used for style transfer task on multiple framework backends.
Module Contents¶
Classes¶
Dataset used for style transfer task on tensorflow/inteltensorflow/tensorflow_itex backend. |
- 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)¶
Bases:
neural_compressor.data.datasets.dataset.DatasetDataset 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.