scripts API

scripts.data_preparation

• scripts.data_preparation.create_lmdb
  • create_lmdb_for_div2k()
  • create_lmdb_for_reds()
  • create_lmdb_for_vimeo90k()
  • prepare_keys_div2k()
  • prepare_keys_reds()
  • prepare_keys_vimeo90k()
• scripts.data_preparation.download_datasets
  • download_dataset()
• scripts.data_preparation.extract_images_from_tfrecords
  • convert_celeba_tfrecords()
  • convert_ffhq_tfrecords()
  • make_ffhq_lmdb_from_imgs()
• scripts.data_preparation.extract_subimages
  • extract_subimages()
  • main()
  • worker()
• scripts.data_preparation.generate_meta_info
  • generate_meta_info_div2k()
• scripts.data_preparation.prepare_hifacegan_dataset
  • DegradationSimulator
    • DegradationSimulator.create_training_dataset()
  • Mosaic16x
    • Mosaic16x.augment_image()
• scripts.data_preparation.regroup_reds_dataset
  • regroup_reds_dataset()

scripts.download_gdrive

• scripts.download_gdrive

scripts.download_pretrained_models

• scripts.download_pretrained_models
  • download_pretrained_models()

scripts.metrics

• scripts.metrics.calculate_fid_folder
  • calculate_fid_folder()
• scripts.metrics.calculate_fid_stats_from_datasets
  • calculate_stats_from_dataset()
• scripts.metrics.calculate_lpips
  • main()
• scripts.metrics.calculate_niqe
  • main()
• scripts.metrics.calculate_psnr_ssim
  • main()
• scripts.metrics.calculate_stylegan2_fid
  • calculate_stylegan2_fid()

scripts.model_conversion

• scripts.model_conversion.convert_dfdnet
  • convert_net()
• scripts.model_conversion.convert_models
  • convert_duf_model()
  • convert_edsr()
  • convert_edvr()
  • convert_esrgan_model()
  • convert_rcan_model()
• scripts.model_conversion.convert_ridnet
• scripts.model_conversion.convert_stylegan
  • convert_net_d()
  • convert_net_g()

scripts.plot

• scripts.plot.model_complexity_cmp_bsrn
  • main()

scripts.publish_models

• scripts.publish_models
  • convert_to_backward_compatible_models()
  • update_sha()