CHANGELOG.md
LICENSE
MANIFEST.in
README.md
pyproject.toml
requirements.txt
examples/__init__.py
examples/bbdm/__init__.py
examples/bbdm/config.py
examples/bbdm/train_bbdm.py
examples/bdbm/__init__.py
examples/bdbm/config.py
examples/bdbm/train_bdbm.py
examples/bibbdm/__init__.py
examples/bibbdm/config.py
examples/bibbdm/train_bibbdm.py
examples/cdtsde/__init__.py
examples/cdtsde/config.py
examples/cdtsde/train_cdtsde.py
examples/community/__init__.py
examples/community/alignflow/__init__.py
examples/community/alignflow/config.py
examples/community/alignflow/pipeline.py
examples/community/alignflow/train.py
examples/community/alignflow/models/__init__.py
examples/community/alignflow/models/cycle_flow/__init__.py
examples/community/alignflow/models/cycle_flow/cycle_flow.py
examples/community/alignflow/models/flow2flow/flow2flow.py
examples/community/alignflow/models/patch_gan/__init__.py
examples/community/alignflow/models/patch_gan/patch_gan.py
examples/community/alignflow/models/real_nvp/__init__.py
examples/community/alignflow/models/real_nvp/coupling_layer.py
examples/community/alignflow/models/real_nvp/real_nvp.py
examples/community/alignflow/models/real_nvp/real_nvp_loss.py
examples/community/alignflow/models/real_nvp/st_resnet.py
examples/community/alignflow/util/__init__.py
examples/community/alignflow/util/array_util.py
examples/community/alignflow/util/gan_util.py
examples/community/alignflow/util/image_util.py
examples/community/alignflow/util/init_util.py
examples/community/alignflow/util/norm_util.py
examples/community/alignflow/util/optim_util.py
examples/community/alignflow/util/shell_util.py
examples/community/bbdm/__init__.py
examples/community/bbdm/convert_ckpt_to_unet.py
examples/community/bbdm/convert_ckpt_to_vqmodel.py
examples/community/bbdm/model.py
examples/community/bbdm/pipeline.py
examples/community/cdtsde/__init__.py
examples/community/cdtsde/__main__.py
examples/community/cdtsde/convert_ckpt_to_cdtsde.py
examples/community/cdtsde/pipeline.py
examples/community/ddbm/__init__.py
examples/community/ddbm/convert_pt_to_unet.py
examples/community/ddbm/model.py
examples/community/ddbm/pipeline.py
examples/community/ddib/__init__.py
examples/community/ddib/convert_pt_to_ddib.py
examples/community/ddib/model.py
examples/community/ddib/pipeline.py
examples/community/diffuseit/__init__.py
examples/community/diffuseit/convert_ckpt_to_diffuseit.py
examples/community/diffuseit/pipeline.py
examples/community/e3diff/__init__.py
examples/community/e3diff/model.py
examples/community/e3diff/pipeline.py
examples/community/e3diff/train.py
examples/community/openearthmap_sar/__init__.py
examples/community/openearthmap_sar/__main__.py
examples/community/openearthmap_sar/model.py
examples/community/openearthmap_sar/pipeline.py
examples/community/openearthmap_sar/train.py
examples/community/parallel_gan/__init__.py
examples/community/parallel_gan/model.py
examples/community/parallel_gan/pipeline.py
examples/community/parallel_gan/train.py
examples/community/sar2optical/__init__.py
examples/community/sar2optical/model.py
examples/community/sar2optical/pipeline.py
examples/community/sar2optical/train.py
examples/community/selfrdb/__init__.py
examples/community/selfrdb/diffusion.py
examples/community/selfrdb/model.py
examples/community/selfrdb/pipeline.py
examples/community/selfrdb/train.py
examples/community/syndiff/__init__.py
examples/community/syndiff/model.py
examples/community/syndiff/pipeline.py
examples/community/syndiff/train.py
examples/cut/__init__.py
examples/cut/config.py
examples/cut/train_cut.py
examples/dbim/__init__.py
examples/dbim/config.py
examples/dbim/train_dbim.py
examples/ddbm/__init__.py
examples/ddbm/config.py
examples/ddbm/train_ddbm.py
examples/ddib/__init__.py
examples/ddib/config.py
examples/ddib/train_ddib.py
examples/i2sb/__init__.py
examples/i2sb/__main__.py
examples/i2sb/config.py
examples/i2sb/trainer.py
examples/lbm/__init__.py
examples/lbm/config.py
examples/lbm/train_lbm.py
examples/lddbm/__init__.py
examples/lddbm/config.py
examples/lddbm/convert_pt_to_lddbm.py
examples/lddbm/train_lddbm.py
examples/local_diffusion/__init__.py
examples/local_diffusion/config.py
examples/local_diffusion/train_local_diffusion.py
examples/pix2pix/__init__.py
examples/pix2pix/train_pix2pix.py
examples/pix2pixhd/__init__.py
examples/pix2pixhd/config.py
examples/pix2pixhd/train_pix2pixhd.py
examples/stargan/__init__.py
examples/stargan/train_stargan.py
examples/stegogan/__init__.py
examples/stegogan/train_stegogan.py
examples/unsb/__init__.py
examples/unsb/config.py
examples/unsb/train_unsb.py
pytorch_image_translation_models.egg-info/PKG-INFO
pytorch_image_translation_models.egg-info/SOURCES.txt
pytorch_image_translation_models.egg-info/dependency_links.txt
pytorch_image_translation_models.egg-info/requires.txt
pytorch_image_translation_models.egg-info/top_level.txt
src/__init__.py
src/data/__init__.py
src/data/datasets.py
src/data/transforms.py
src/losses/__init__.py
src/losses/adversarial.py
src/losses/diffusion.py
src/losses/perceptual.py
src/metrics/__init__.py
src/metrics/conditional/__init__.py
src/metrics/conditional/_inception.py
src/metrics/conditional/afd.py
src/metrics/conditional/evaluator.py
src/metrics/conditional/registry.py
src/metrics/conditional/vs.py
src/metrics/paired/__init__.py
src/metrics/paired/dists_impl.py
src/metrics/paired/evaluator.py
src/metrics/paired/l1_l2.py
src/metrics/paired/lpips_impl.py
src/metrics/paired/psnr.py
src/metrics/paired/registry.py
src/metrics/paired/samscore_impl.py
src/metrics/paired/ssim.py
src/metrics/unpaired/__init__.py
src/metrics/unpaired/cmmd.py
src/metrics/unpaired/evaluator.py
src/metrics/unpaired/fid.py
src/metrics/unpaired/fwd.py
src/metrics/unpaired/ifid.py
src/metrics/unpaired/is_.py
src/metrics/unpaired/kid.py
src/metrics/unpaired/precision_recall.py
src/metrics/unpaired/registry.py
src/metrics/unpaired/sfd.py
src/metrics/unpaired/sfid.py
src/models/__init__.py
src/models/cut/__init__.py
src/models/cut/cut_model.py
src/models/discriminators/__init__.py
src/models/discriminators/patchgan.py
src/models/dit/__init__.py
src/models/dit/sit.py
src/models/ecsi/diffusion.py
src/models/ecsi/model.py
src/models/ecsi/route.py
src/models/ecsi/sampling.py
src/models/ecsi/_utils/edm_unet.py
src/models/ecsi/_utils/fp16_util.py
src/models/ecsi/_utils/nn.py
src/models/ecsi/_utils/unet.py
src/models/fcdm/__init__.py
src/models/fcdm/fcdm_models.py
src/models/generators/__init__.py
src/models/generators/resnet.py
src/models/generators/unet.py
src/models/lddbm/__init__.py
src/models/lddbm/bridge_model.py
src/models/lddbm/diffusion_utils.py
src/models/lddbm/encoder_decoder.py
src/models/lddbm/karras_diffusion.py
src/models/lddbm/mm_dist_trans.py
src/models/lddbm/models_loader.py
src/models/lddbm/names.py
src/models/lddbm/nn_utils.py
src/models/lddbm/resample.py
src/models/lddbm/transformer.py
src/models/lddbm/vae_modules.py
src/models/local_diffusion/__init__.py
src/models/local_diffusion/local_diffusion_model.py
src/models/pix2pixhd/__init__.py
src/models/pix2pixhd/blocks.py
src/models/pix2pixhd/generator.py
src/models/stargan/__init__.py
src/models/stargan/blocks.py
src/models/stargan/discriminator.py
src/models/stargan/generator.py
src/models/stegogan/__init__.py
src/models/stegogan/generators.py
src/models/stegogan/networks.py
src/models/unet/__init__.py
src/models/unet/adm.py
src/models/unet/edm.py
src/models/unet/rin.py
src/models/unet/sid.py
src/models/unet/sid2.py
src/models/unet/vdmpp.py
src/models/unsb/__init__.py
src/models/unsb/unsb_model.py
src/pipelines/__init__.py
src/pipelines/bbdm.py
src/pipelines/bdbm.py
src/pipelines/bibbdm.py
src/pipelines/cdtsde.py
src/pipelines/cut.py
src/pipelines/dbim.py
src/pipelines/ddbm.py
src/pipelines/ddib.py
src/pipelines/ecsi.py
src/pipelines/fcdm.py
src/pipelines/i2sb.py
src/pipelines/lbm.py
src/pipelines/lddbm.py
src/pipelines/local_diffusion.py
src/pipelines/pix2pix.py
src/pipelines/pix2pixhd.py
src/pipelines/stargan.py
src/pipelines/stegogan.py
src/pipelines/unsb.py
src/schedulers/__init__.py
src/schedulers/bbdm.py
src/schedulers/bdbm.py
src/schedulers/bibbdm.py
src/schedulers/cdtsde.py
src/schedulers/dbim.py
src/schedulers/ddbm.py
src/schedulers/ddib.py
src/schedulers/fcdm.py
src/schedulers/i2sb.py
src/schedulers/lbm.py
src/schedulers/local_diffusion.py
src/schedulers/sid2.py
src/schedulers/unsb.py
src/training/__init__.py
src/training/argparse_utils.py
src/training/checkpoint.py
src/training/common.py
src/training/optimizer_utils.py
src/utils/__init__.py
src/utils/config_yaml.py
src/utils/training_utils.py
tests/test_community_pipelines.py
tests/test_data.py
tests/test_dit_lbm.py
tests/test_i2sb_baseline.py
tests/test_inference.py
tests/test_local_diffusion.py
tests/test_losses.py
tests/test_metrics.py
tests/test_metrics_lpips_dists.py
tests/test_metrics_unpaired.py
tests/test_models.py
tests/test_new_schedulers_pipelines.py
tests/test_pipeline_examples.py
tests/test_stargan.py
tests/test_stegogan.py
tests/test_unsb.py