├── .gitignore ├── LICENSE ├── README.md ├── assets ├── fig_framework.jpg └── fig_teaser.jpg ├── configs ├── 256_codiff_mask_text.yaml ├── 256_mask.yaml ├── 256_text.yaml ├── 256_vae.yaml ├── 512_codiff_mask_text.yaml ├── 512_mask.yaml ├── 512_text.yaml └── 512_vae.yaml ├── editing ├── collaborative_edit.py ├── imagic_edit_mask.py └── imagic_edit_text.py ├── environment.yaml ├── freeu ├── README.md ├── assets │ ├── mask2face_27007.jpeg │ ├── mask2face_29980.jpeg │ ├── text2face_female.jpeg │ └── text2face_male.jpeg ├── mask2image_freeu.py └── text2image_freeu.py ├── generate.py ├── ldm ├── data │ ├── __init__.py │ ├── base.py │ ├── celebahq.py │ ├── imagenet.py │ └── lsun.py ├── lr_scheduler.py ├── models │ ├── autoencoder.py │ └── diffusion │ │ ├── __init__.py │ │ ├── classifier.py │ │ ├── compose_modules.py │ │ ├── ddim.py │ │ ├── ddim_confidence.py │ │ ├── ddpm.py │ │ ├── ddpm_compose.py │ │ └── plms.py ├── modules │ ├── attention.py │ ├── diffusionmodules │ │ ├── __init__.py │ │ ├── model.py │ │ ├── openaimodel.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 │ ├── losses │ │ ├── __init__.py │ │ ├── contperceptual.py │ │ └── vqperceptual.py │ └── x_transformer.py └── util.py ├── main.py ├── mask2image.py ├── setup.py ├── test_data ├── 256_masks │ └── 29980.png ├── 512_masks │ ├── 27007.png │ └── 29980.png └── test_mask_edit │ ├── 256_edited_masks │ └── 27044_0_remove_smile_and_rings.png │ └── 256_input_image │ └── 27044.jpg └── text2image.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/README.md -------------------------------------------------------------------------------- /assets/fig_framework.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/assets/fig_framework.jpg -------------------------------------------------------------------------------- /assets/fig_teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/assets/fig_teaser.jpg -------------------------------------------------------------------------------- /configs/256_codiff_mask_text.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/configs/256_codiff_mask_text.yaml -------------------------------------------------------------------------------- /configs/256_mask.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/configs/256_mask.yaml -------------------------------------------------------------------------------- /configs/256_text.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/configs/256_text.yaml -------------------------------------------------------------------------------- /configs/256_vae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/configs/256_vae.yaml -------------------------------------------------------------------------------- /configs/512_codiff_mask_text.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/configs/512_codiff_mask_text.yaml -------------------------------------------------------------------------------- /configs/512_mask.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/configs/512_mask.yaml -------------------------------------------------------------------------------- /configs/512_text.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/configs/512_text.yaml -------------------------------------------------------------------------------- /configs/512_vae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/configs/512_vae.yaml -------------------------------------------------------------------------------- /editing/collaborative_edit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/editing/collaborative_edit.py -------------------------------------------------------------------------------- /editing/imagic_edit_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/editing/imagic_edit_mask.py -------------------------------------------------------------------------------- /editing/imagic_edit_text.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/editing/imagic_edit_text.py -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/environment.yaml -------------------------------------------------------------------------------- /freeu/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/freeu/README.md -------------------------------------------------------------------------------- /freeu/assets/mask2face_27007.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/freeu/assets/mask2face_27007.jpeg -------------------------------------------------------------------------------- /freeu/assets/mask2face_29980.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/freeu/assets/mask2face_29980.jpeg -------------------------------------------------------------------------------- /freeu/assets/text2face_female.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/freeu/assets/text2face_female.jpeg -------------------------------------------------------------------------------- /freeu/assets/text2face_male.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/freeu/assets/text2face_male.jpeg -------------------------------------------------------------------------------- /freeu/mask2image_freeu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/freeu/mask2image_freeu.py -------------------------------------------------------------------------------- /freeu/text2image_freeu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/freeu/text2image_freeu.py -------------------------------------------------------------------------------- /generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/generate.py -------------------------------------------------------------------------------- /ldm/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/data/base.py -------------------------------------------------------------------------------- /ldm/data/celebahq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/data/celebahq.py -------------------------------------------------------------------------------- /ldm/data/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/data/imagenet.py -------------------------------------------------------------------------------- /ldm/data/lsun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/data/lsun.py -------------------------------------------------------------------------------- /ldm/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/lr_scheduler.py -------------------------------------------------------------------------------- /ldm/models/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/models/autoencoder.py -------------------------------------------------------------------------------- /ldm/models/diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/models/diffusion/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/models/diffusion/classifier.py -------------------------------------------------------------------------------- /ldm/models/diffusion/compose_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/models/diffusion/compose_modules.py -------------------------------------------------------------------------------- /ldm/models/diffusion/ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/models/diffusion/ddim.py -------------------------------------------------------------------------------- /ldm/models/diffusion/ddim_confidence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/models/diffusion/ddim_confidence.py -------------------------------------------------------------------------------- /ldm/models/diffusion/ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/models/diffusion/ddpm.py -------------------------------------------------------------------------------- /ldm/models/diffusion/ddpm_compose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/models/diffusion/ddpm_compose.py -------------------------------------------------------------------------------- /ldm/models/diffusion/plms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/models/diffusion/plms.py -------------------------------------------------------------------------------- /ldm/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/modules/attention.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/modules/diffusionmodules/model.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/openaimodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/modules/diffusionmodules/openaimodel.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/modules/diffusionmodules/util.py -------------------------------------------------------------------------------- /ldm/modules/distributions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/distributions/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/modules/distributions/distributions.py -------------------------------------------------------------------------------- /ldm/modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/modules/ema.py -------------------------------------------------------------------------------- /ldm/modules/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/encoders/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/modules/encoders/modules.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/modules/image_degradation/__init__.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/bsrgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/modules/image_degradation/bsrgan.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/bsrgan_light.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/modules/image_degradation/bsrgan_light.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/utils/test.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/modules/image_degradation/utils/test.png -------------------------------------------------------------------------------- /ldm/modules/image_degradation/utils_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/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/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/modules/losses/contperceptual.py -------------------------------------------------------------------------------- /ldm/modules/losses/vqperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/modules/losses/vqperceptual.py -------------------------------------------------------------------------------- /ldm/modules/x_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/modules/x_transformer.py -------------------------------------------------------------------------------- /ldm/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/ldm/util.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/main.py -------------------------------------------------------------------------------- /mask2image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/mask2image.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/setup.py -------------------------------------------------------------------------------- /test_data/256_masks/29980.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/test_data/256_masks/29980.png -------------------------------------------------------------------------------- /test_data/512_masks/27007.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/test_data/512_masks/27007.png -------------------------------------------------------------------------------- /test_data/512_masks/29980.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/test_data/512_masks/29980.png -------------------------------------------------------------------------------- /test_data/test_mask_edit/256_edited_masks/27044_0_remove_smile_and_rings.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/test_data/test_mask_edit/256_edited_masks/27044_0_remove_smile_and_rings.png -------------------------------------------------------------------------------- /test_data/test_mask_edit/256_input_image/27044.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/test_data/test_mask_edit/256_input_image/27044.jpg -------------------------------------------------------------------------------- /text2image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziqihuangg/Collaborative-Diffusion/HEAD/text2image.py --------------------------------------------------------------------------------