scripts.data_preparation.extract_images_from_tfrecords

scripts.data_preparation.extract_images_from_tfrecords.convert_celeba_tfrecords(tf_file, log_resolution, save_root, save_type='img', compress_level=1)

Convert CelebA tfrecords to images or lmdb files.

Parameters:

• tf_file (str) – Input tfrecords file in glob pattern. Example: ‘datasets/celeba/celeba_tfrecords/validation/validation-r08-s--of-.tfrecords’ # noqa:E501

• log_resolution (int) – Log scale of resolution.

• save_root (str) – Path root to save.

• save_type (str) – Save type. Options: img | lmdb. Default: img.

• compress_level (int) – Compress level when encoding images. Default: 1.

scripts.data_preparation.extract_images_from_tfrecords.convert_ffhq_tfrecords(tf_file, log_resolution, save_root, save_type='img', compress_level=1)

Convert FFHQ tfrecords to images or lmdb files.

Parameters:

• tf_file (str) – Input tfrecords file.

• log_resolution (int) – Log scale of resolution.

• save_root (str) – Path root to save.

• save_type (str) – Save type. Options: img | lmdb. Default: img.

• compress_level (int) – Compress level when encoding images. Default: 1.

scripts.data_preparation.extract_images_from_tfrecords.make_ffhq_lmdb_from_imgs(folder_path, log_resolution, save_root, save_type='lmdb', compress_level=1)

Make FFHQ lmdb from images.

Parameters:

• folder_path (str) – Folder path.

• log_resolution (int) – Log scale of resolution.

• save_root (str) – Path root to save.

• save_type (str) – Save type. Options: img | lmdb. Default: img.

• compress_level (int) – Compress level when encoding images. Default: 1.