scripts.data_preparation.prepare_hifacegan_dataset

class scripts.data_preparation.prepare_hifacegan_dataset.DegradationSimulator[source]

Bases: object

Generating training/testing data pairs on the fly. The degradation script is aligned with HiFaceGAN paper settings.

Parameters:

  • opt (str | op) – Config for degradation script, with degradation type and parameters

  • ia.augmentors (Custom degradation is possible by passing an inherited class from) –

create_training_dataset(deg, gt_folder, lq_folder=None)[source]
class scripts.data_preparation.prepare_hifacegan_dataset.Mosaic16x[source]

Bases: object

Mosaic16x: A customized image augmentor for 16-pixel mosaic By default it replaces each pixel value with the mean value of its 16x16 neighborhood

augment_image(x)[source]