├── LICENSE ├── PID.png ├── README.md ├── TeVNet ├── README.md ├── datasets.py ├── models.py ├── shell │ ├── test_tev2hsv.sh │ ├── test_vnums4.sh │ └── train_vnums4.sh ├── test.py ├── tev2hsv.py ├── train.py └── utils.py ├── configs ├── autoencoder │ ├── autoencoder_kl_16x16x16.yaml │ ├── autoencoder_kl_32x32x4.yaml │ ├── autoencoder_kl_64x64x3.yaml │ └── autoencoder_kl_8x8x64.yaml ├── latent-diffusion │ ├── PID-FLIR-c=E.yaml │ ├── PID-FLIR-c=M.yaml │ ├── PID-KAIST-c=E.yaml │ ├── PID-KAIST-c=M.yaml │ └── PID-VEDAI-c=E.yaml └── retrieval-augmented-diffusion │ └── 768x768.yaml ├── data ├── FLIRv1512 │ ├── FLIR_512_train.txt │ └── FLIR_512_val.txt ├── KAIST512 │ ├── KAIST_512_test.txt │ └── KAIST_512_train.txt └── VEDAI512 │ ├── test.txt │ └── train.txt ├── data_alignment └── FLIRv1_preprocess.py ├── dataset └── KAIST512 │ ├── lwir-gt │ ├── set07_V001_I00039.png │ ├── set08_V001_I01239.png │ ├── set09_V000_I01239.png │ ├── set09_V000_I03159.png │ └── set11_V001_I01199.png │ └── visible │ ├── set07_V001_I00039.png │ ├── set08_V001_I01239.png │ ├── set09_V000_I01239.png │ ├── set09_V000_I03159.png │ └── set11_V001_I01199.png ├── environment.yaml ├── ldm ├── data │ ├── FLIRv1.py │ ├── KAIST.py │ ├── __init__.py │ ├── base.py │ └── vedai512.py ├── lr_scheduler.py ├── models │ ├── autoencoder.py │ └── diffusion │ │ ├── HADARloss.py │ │ ├── __init__.py │ │ ├── classifier.py │ │ ├── ddim.py │ │ ├── ddpm.py │ │ ├── ddpm_tev.py │ │ └── plms.py ├── modules │ ├── HADARNet │ │ ├── __init__.py │ │ └── modules.py │ ├── attention.py │ ├── diffusionmodules │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-38.pyc │ │ │ ├── model.cpython-38.pyc │ │ │ ├── openaimodel.cpython-38.pyc │ │ │ └── util.cpython-38.pyc │ │ ├── model.py │ │ ├── openaimodel.py │ │ └── util.py │ ├── distributions │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-38.pyc │ │ │ └── distributions.cpython-38.pyc │ │ └── distributions.py │ ├── ema.py │ ├── encoders │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-38.pyc │ │ │ └── modules.cpython-38.pyc │ │ └── modules.py │ ├── image_degradation │ │ ├── __init__.py │ │ ├── bsrgan.py │ │ ├── bsrgan_light.py │ │ ├── utils │ │ │ └── test.png │ │ └── utils_image.py │ ├── losses │ │ ├── __init__.py │ │ ├── contperceptual.py │ │ └── vqperceptual.py │ └── x_transformer.py └── util.py ├── main.py ├── metric ├── core │ └── metrics.py └── eval.py ├── models └── first_stage_models │ ├── kl-f16 │ └── config.yaml │ ├── kl-f32 │ └── config.yaml │ ├── kl-f4 │ └── config.yaml │ ├── kl-f8 │ └── config.yaml │ ├── vq-f16 │ └── config.yaml │ ├── vq-f4-noattn │ └── config.yaml │ ├── vq-f4 │ └── config.yaml │ ├── vq-f8-n256 │ └── config.yaml │ └── vq-f8 │ └── config.yaml ├── scripts └── rgb2ir_vqf8.py ├── shell ├── run_test_kaist512.sh └── run_train_kaist512.sh └── taming ├── lr_scheduler.py ├── models ├── cond_transformer.py ├── dummy_cond_stage.py └── vqgan.py ├── modules └── misc │ └── coord.py └── util.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/LICENSE -------------------------------------------------------------------------------- /PID.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/PID.png -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/README.md -------------------------------------------------------------------------------- /TeVNet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/TeVNet/README.md -------------------------------------------------------------------------------- /TeVNet/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/TeVNet/datasets.py -------------------------------------------------------------------------------- /TeVNet/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/TeVNet/models.py -------------------------------------------------------------------------------- /TeVNet/shell/test_tev2hsv.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/TeVNet/shell/test_tev2hsv.sh -------------------------------------------------------------------------------- /TeVNet/shell/test_vnums4.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/TeVNet/shell/test_vnums4.sh -------------------------------------------------------------------------------- /TeVNet/shell/train_vnums4.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/TeVNet/shell/train_vnums4.sh -------------------------------------------------------------------------------- /TeVNet/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/TeVNet/test.py -------------------------------------------------------------------------------- /TeVNet/tev2hsv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/TeVNet/tev2hsv.py -------------------------------------------------------------------------------- /TeVNet/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/TeVNet/train.py -------------------------------------------------------------------------------- /TeVNet/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/TeVNet/utils.py -------------------------------------------------------------------------------- /configs/autoencoder/autoencoder_kl_16x16x16.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/configs/autoencoder/autoencoder_kl_16x16x16.yaml -------------------------------------------------------------------------------- /configs/autoencoder/autoencoder_kl_32x32x4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/configs/autoencoder/autoencoder_kl_32x32x4.yaml -------------------------------------------------------------------------------- /configs/autoencoder/autoencoder_kl_64x64x3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/configs/autoencoder/autoencoder_kl_64x64x3.yaml -------------------------------------------------------------------------------- /configs/autoencoder/autoencoder_kl_8x8x64.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/configs/autoencoder/autoencoder_kl_8x8x64.yaml -------------------------------------------------------------------------------- /configs/latent-diffusion/PID-FLIR-c=E.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/configs/latent-diffusion/PID-FLIR-c=E.yaml -------------------------------------------------------------------------------- /configs/latent-diffusion/PID-FLIR-c=M.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/configs/latent-diffusion/PID-FLIR-c=M.yaml -------------------------------------------------------------------------------- /configs/latent-diffusion/PID-KAIST-c=E.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/configs/latent-diffusion/PID-KAIST-c=E.yaml -------------------------------------------------------------------------------- /configs/latent-diffusion/PID-KAIST-c=M.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/configs/latent-diffusion/PID-KAIST-c=M.yaml -------------------------------------------------------------------------------- /configs/latent-diffusion/PID-VEDAI-c=E.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/configs/latent-diffusion/PID-VEDAI-c=E.yaml -------------------------------------------------------------------------------- /configs/retrieval-augmented-diffusion/768x768.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/configs/retrieval-augmented-diffusion/768x768.yaml -------------------------------------------------------------------------------- /data/FLIRv1512/FLIR_512_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/data/FLIRv1512/FLIR_512_train.txt -------------------------------------------------------------------------------- /data/FLIRv1512/FLIR_512_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/data/FLIRv1512/FLIR_512_val.txt -------------------------------------------------------------------------------- /data/KAIST512/KAIST_512_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/data/KAIST512/KAIST_512_test.txt -------------------------------------------------------------------------------- /data/KAIST512/KAIST_512_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/data/KAIST512/KAIST_512_train.txt -------------------------------------------------------------------------------- /data/VEDAI512/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/data/VEDAI512/test.txt -------------------------------------------------------------------------------- /data/VEDAI512/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/data/VEDAI512/train.txt -------------------------------------------------------------------------------- /data_alignment/FLIRv1_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/data_alignment/FLIRv1_preprocess.py -------------------------------------------------------------------------------- /dataset/KAIST512/lwir-gt/set07_V001_I00039.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/dataset/KAIST512/lwir-gt/set07_V001_I00039.png -------------------------------------------------------------------------------- /dataset/KAIST512/lwir-gt/set08_V001_I01239.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/dataset/KAIST512/lwir-gt/set08_V001_I01239.png -------------------------------------------------------------------------------- /dataset/KAIST512/lwir-gt/set09_V000_I01239.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/dataset/KAIST512/lwir-gt/set09_V000_I01239.png -------------------------------------------------------------------------------- /dataset/KAIST512/lwir-gt/set09_V000_I03159.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/dataset/KAIST512/lwir-gt/set09_V000_I03159.png -------------------------------------------------------------------------------- /dataset/KAIST512/lwir-gt/set11_V001_I01199.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/dataset/KAIST512/lwir-gt/set11_V001_I01199.png -------------------------------------------------------------------------------- /dataset/KAIST512/visible/set07_V001_I00039.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/dataset/KAIST512/visible/set07_V001_I00039.png -------------------------------------------------------------------------------- /dataset/KAIST512/visible/set08_V001_I01239.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/dataset/KAIST512/visible/set08_V001_I01239.png -------------------------------------------------------------------------------- /dataset/KAIST512/visible/set09_V000_I01239.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/dataset/KAIST512/visible/set09_V000_I01239.png -------------------------------------------------------------------------------- /dataset/KAIST512/visible/set09_V000_I03159.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/dataset/KAIST512/visible/set09_V000_I03159.png -------------------------------------------------------------------------------- /dataset/KAIST512/visible/set11_V001_I01199.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/dataset/KAIST512/visible/set11_V001_I01199.png -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/environment.yaml -------------------------------------------------------------------------------- /ldm/data/FLIRv1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/data/FLIRv1.py -------------------------------------------------------------------------------- /ldm/data/KAIST.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/data/KAIST.py -------------------------------------------------------------------------------- /ldm/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/data/base.py -------------------------------------------------------------------------------- /ldm/data/vedai512.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/data/vedai512.py -------------------------------------------------------------------------------- /ldm/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/lr_scheduler.py -------------------------------------------------------------------------------- /ldm/models/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/models/autoencoder.py -------------------------------------------------------------------------------- /ldm/models/diffusion/HADARloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/models/diffusion/HADARloss.py -------------------------------------------------------------------------------- /ldm/models/diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/models/diffusion/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/models/diffusion/classifier.py -------------------------------------------------------------------------------- /ldm/models/diffusion/ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/models/diffusion/ddim.py -------------------------------------------------------------------------------- /ldm/models/diffusion/ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/models/diffusion/ddpm.py -------------------------------------------------------------------------------- /ldm/models/diffusion/ddpm_tev.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/models/diffusion/ddpm_tev.py -------------------------------------------------------------------------------- /ldm/models/diffusion/plms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/models/diffusion/plms.py -------------------------------------------------------------------------------- /ldm/modules/HADARNet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/HADARNet/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/modules/HADARNet/modules.py -------------------------------------------------------------------------------- /ldm/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/modules/attention.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/modules/diffusionmodules/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/__pycache__/model.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/modules/diffusionmodules/__pycache__/model.cpython-38.pyc -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/__pycache__/openaimodel.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/modules/diffusionmodules/__pycache__/openaimodel.cpython-38.pyc -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/__pycache__/util.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/modules/diffusionmodules/__pycache__/util.cpython-38.pyc -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/modules/diffusionmodules/model.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/openaimodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/modules/diffusionmodules/openaimodel.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/modules/diffusionmodules/util.py -------------------------------------------------------------------------------- /ldm/modules/distributions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/distributions/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/modules/distributions/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /ldm/modules/distributions/__pycache__/distributions.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/modules/distributions/__pycache__/distributions.cpython-38.pyc -------------------------------------------------------------------------------- /ldm/modules/distributions/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/modules/distributions/distributions.py -------------------------------------------------------------------------------- /ldm/modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/modules/ema.py -------------------------------------------------------------------------------- /ldm/modules/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/encoders/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/modules/encoders/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /ldm/modules/encoders/__pycache__/modules.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/modules/encoders/__pycache__/modules.cpython-38.pyc -------------------------------------------------------------------------------- /ldm/modules/encoders/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/modules/encoders/modules.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/modules/image_degradation/__init__.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/bsrgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/modules/image_degradation/bsrgan.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/bsrgan_light.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/modules/image_degradation/bsrgan_light.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/utils/test.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/modules/image_degradation/utils/test.png -------------------------------------------------------------------------------- /ldm/modules/image_degradation/utils_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/modules/image_degradation/utils_image.py -------------------------------------------------------------------------------- /ldm/modules/losses/__init__.py: -------------------------------------------------------------------------------- 1 | from ldm.modules.losses.contperceptual import LPIPSWithDiscriminator -------------------------------------------------------------------------------- /ldm/modules/losses/contperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/modules/losses/contperceptual.py -------------------------------------------------------------------------------- /ldm/modules/losses/vqperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/modules/losses/vqperceptual.py -------------------------------------------------------------------------------- /ldm/modules/x_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/modules/x_transformer.py -------------------------------------------------------------------------------- /ldm/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/ldm/util.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/main.py -------------------------------------------------------------------------------- /metric/core/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/metric/core/metrics.py -------------------------------------------------------------------------------- /metric/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/metric/eval.py -------------------------------------------------------------------------------- /models/first_stage_models/kl-f16/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/models/first_stage_models/kl-f16/config.yaml -------------------------------------------------------------------------------- /models/first_stage_models/kl-f32/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/models/first_stage_models/kl-f32/config.yaml -------------------------------------------------------------------------------- /models/first_stage_models/kl-f4/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/models/first_stage_models/kl-f4/config.yaml -------------------------------------------------------------------------------- /models/first_stage_models/kl-f8/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/models/first_stage_models/kl-f8/config.yaml -------------------------------------------------------------------------------- /models/first_stage_models/vq-f16/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/models/first_stage_models/vq-f16/config.yaml -------------------------------------------------------------------------------- /models/first_stage_models/vq-f4-noattn/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/models/first_stage_models/vq-f4-noattn/config.yaml -------------------------------------------------------------------------------- /models/first_stage_models/vq-f4/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/models/first_stage_models/vq-f4/config.yaml -------------------------------------------------------------------------------- /models/first_stage_models/vq-f8-n256/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/models/first_stage_models/vq-f8-n256/config.yaml -------------------------------------------------------------------------------- /models/first_stage_models/vq-f8/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/models/first_stage_models/vq-f8/config.yaml -------------------------------------------------------------------------------- /scripts/rgb2ir_vqf8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/scripts/rgb2ir_vqf8.py -------------------------------------------------------------------------------- /shell/run_test_kaist512.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/shell/run_test_kaist512.sh -------------------------------------------------------------------------------- /shell/run_train_kaist512.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/shell/run_train_kaist512.sh -------------------------------------------------------------------------------- /taming/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/taming/lr_scheduler.py -------------------------------------------------------------------------------- /taming/models/cond_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/taming/models/cond_transformer.py -------------------------------------------------------------------------------- /taming/models/dummy_cond_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/taming/models/dummy_cond_stage.py -------------------------------------------------------------------------------- /taming/models/vqgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/taming/models/vqgan.py -------------------------------------------------------------------------------- /taming/modules/misc/coord.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/taming/modules/misc/coord.py -------------------------------------------------------------------------------- /taming/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fangyuanmao/PID/HEAD/taming/util.py --------------------------------------------------------------------------------