├── README.md ├── attention_processor.py ├── base_network.py ├── ber.py ├── ckpts_cls └── config.yml ├── ckpts_reg └── config.yml ├── cldm ├── GIFT.py ├── cldm.py ├── ddim_hacked.py ├── encoder_res18.py ├── hack.py ├── logger.py └── model.py ├── eval.py ├── figures ├── cmp_with_SGDiffusion.jpg ├── cmp_with_baselines.jpg └── readme.txt ├── infer_GPSDiffusion.py ├── infer_single.py ├── ip_adapter.py ├── ldm ├── models │ ├── autoencoder.py │ └── diffusion │ │ ├── __init__.py │ │ ├── ddim.py │ │ ├── ddpm.py │ │ ├── dpm_solver │ │ ├── __init__.py │ │ ├── dpm_solver.py │ │ └── sampler.py │ │ ├── plms.py │ │ └── sampling_util.py ├── modules │ ├── attention.py │ ├── diffusionmodules │ │ ├── __init__.py │ │ ├── model.py │ │ ├── openaimodel.py │ │ ├── upscaling.py │ │ └── util.py │ ├── distributions │ │ ├── __init__.py │ │ └── distributions.py │ ├── ema.py │ ├── encoders │ │ ├── __init__.py │ │ └── modules.py │ ├── image_degradation │ │ ├── __init__.py │ │ ├── bsrgan.py │ │ ├── bsrgan_light.py │ │ ├── utils │ │ │ └── test.png │ │ └── utils_image.py │ └── midas │ │ ├── __init__.py │ │ ├── api.py │ │ ├── midas │ │ ├── __init__.py │ │ ├── base_model.py │ │ ├── blocks.py │ │ ├── dpt_depth.py │ │ ├── midas_net.py │ │ ├── midas_net_custom.py │ │ ├── transforms.py │ │ └── vit.py │ │ └── utils.py └── util.py ├── models ├── cldm_v15.yaml ├── cldm_v21.yaml └── shadow_clusters.txt ├── post_processing.py ├── requirements.txt ├── results.jpg ├── shadow_clusters.txt ├── src ├── bbx_network.py ├── bbx_reg.py ├── config.py ├── datasets_bbx.py ├── datasets_cls.py ├── loss.py ├── mask_cls.py └── model.py ├── ssim.py ├── train_GPSDiffusion.py ├── train_bbx.py ├── train_cls.py ├── train_post_process.py ├── tutorial_dataset.py └── util.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/README.md -------------------------------------------------------------------------------- /attention_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/attention_processor.py -------------------------------------------------------------------------------- /base_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/base_network.py -------------------------------------------------------------------------------- /ber.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ber.py -------------------------------------------------------------------------------- /ckpts_cls/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ckpts_cls/config.yml -------------------------------------------------------------------------------- /ckpts_reg/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ckpts_reg/config.yml -------------------------------------------------------------------------------- /cldm/GIFT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/cldm/GIFT.py -------------------------------------------------------------------------------- /cldm/cldm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/cldm/cldm.py -------------------------------------------------------------------------------- /cldm/ddim_hacked.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/cldm/ddim_hacked.py -------------------------------------------------------------------------------- /cldm/encoder_res18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/cldm/encoder_res18.py -------------------------------------------------------------------------------- /cldm/hack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/cldm/hack.py -------------------------------------------------------------------------------- /cldm/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/cldm/logger.py -------------------------------------------------------------------------------- /cldm/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/cldm/model.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/eval.py -------------------------------------------------------------------------------- /figures/cmp_with_SGDiffusion.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/figures/cmp_with_SGDiffusion.jpg -------------------------------------------------------------------------------- /figures/cmp_with_baselines.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/figures/cmp_with_baselines.jpg -------------------------------------------------------------------------------- /figures/readme.txt: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /infer_GPSDiffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/infer_GPSDiffusion.py -------------------------------------------------------------------------------- /infer_single.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/infer_single.py -------------------------------------------------------------------------------- /ip_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ip_adapter.py -------------------------------------------------------------------------------- /ldm/models/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/models/autoencoder.py -------------------------------------------------------------------------------- /ldm/models/diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/models/diffusion/ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/models/diffusion/ddim.py -------------------------------------------------------------------------------- /ldm/models/diffusion/ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/models/diffusion/ddpm.py -------------------------------------------------------------------------------- /ldm/models/diffusion/dpm_solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/models/diffusion/dpm_solver/__init__.py -------------------------------------------------------------------------------- /ldm/models/diffusion/dpm_solver/dpm_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/models/diffusion/dpm_solver/dpm_solver.py -------------------------------------------------------------------------------- /ldm/models/diffusion/dpm_solver/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/models/diffusion/dpm_solver/sampler.py -------------------------------------------------------------------------------- /ldm/models/diffusion/plms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/models/diffusion/plms.py -------------------------------------------------------------------------------- /ldm/models/diffusion/sampling_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/models/diffusion/sampling_util.py -------------------------------------------------------------------------------- /ldm/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/modules/attention.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/modules/diffusionmodules/model.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/openaimodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/modules/diffusionmodules/openaimodel.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/upscaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/modules/diffusionmodules/upscaling.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/modules/diffusionmodules/util.py -------------------------------------------------------------------------------- /ldm/modules/distributions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/distributions/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/modules/distributions/distributions.py -------------------------------------------------------------------------------- /ldm/modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/modules/ema.py -------------------------------------------------------------------------------- /ldm/modules/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/encoders/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/modules/encoders/modules.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/modules/image_degradation/__init__.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/bsrgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/modules/image_degradation/bsrgan.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/bsrgan_light.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/modules/image_degradation/bsrgan_light.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/utils/test.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/modules/image_degradation/utils/test.png -------------------------------------------------------------------------------- /ldm/modules/image_degradation/utils_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/modules/image_degradation/utils_image.py -------------------------------------------------------------------------------- /ldm/modules/midas/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/midas/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/modules/midas/api.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/midas/midas/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/modules/midas/midas/base_model.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/modules/midas/midas/blocks.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/dpt_depth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/modules/midas/midas/dpt_depth.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/midas_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/modules/midas/midas/midas_net.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/midas_net_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/modules/midas/midas/midas_net_custom.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/modules/midas/midas/transforms.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/modules/midas/midas/vit.py -------------------------------------------------------------------------------- /ldm/modules/midas/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/modules/midas/utils.py -------------------------------------------------------------------------------- /ldm/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ldm/util.py -------------------------------------------------------------------------------- /models/cldm_v15.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/models/cldm_v15.yaml -------------------------------------------------------------------------------- /models/cldm_v21.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/models/cldm_v21.yaml -------------------------------------------------------------------------------- /models/shadow_clusters.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/models/shadow_clusters.txt -------------------------------------------------------------------------------- /post_processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/post_processing.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/requirements.txt -------------------------------------------------------------------------------- /results.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/results.jpg -------------------------------------------------------------------------------- /shadow_clusters.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/shadow_clusters.txt -------------------------------------------------------------------------------- /src/bbx_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/src/bbx_network.py -------------------------------------------------------------------------------- /src/bbx_reg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/src/bbx_reg.py -------------------------------------------------------------------------------- /src/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/src/config.py -------------------------------------------------------------------------------- /src/datasets_bbx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/src/datasets_bbx.py -------------------------------------------------------------------------------- /src/datasets_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/src/datasets_cls.py -------------------------------------------------------------------------------- /src/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/src/loss.py -------------------------------------------------------------------------------- /src/mask_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/src/mask_cls.py -------------------------------------------------------------------------------- /src/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/src/model.py -------------------------------------------------------------------------------- /ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/ssim.py -------------------------------------------------------------------------------- /train_GPSDiffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/train_GPSDiffusion.py -------------------------------------------------------------------------------- /train_bbx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/train_bbx.py -------------------------------------------------------------------------------- /train_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/train_cls.py -------------------------------------------------------------------------------- /train_post_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/train_post_process.py -------------------------------------------------------------------------------- /tutorial_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/tutorial_dataset.py -------------------------------------------------------------------------------- /util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/GPSDiffusion-Object-Shadow-Generation/HEAD/util.py --------------------------------------------------------------------------------