├── .gitignore ├── LICENSE ├── README.md ├── configs ├── .gitignore ├── ablation │ ├── celeba_small.yml │ └── pretrain │ │ ├── block2 │ │ ├── celeba_10p.yml │ │ ├── celeba_50p.yml │ │ └── celeba_90p.yml │ │ ├── block4 │ │ ├── celeba_10p.yml │ │ ├── celeba_50p.yml │ │ └── celeba_90p.yml │ │ ├── block8 │ │ └── celeba_50p.yml │ │ ├── crop │ │ ├── celeba_10p.yml │ │ ├── celeba_50p.yml │ │ └── celeba_90p.yml │ │ └── patch │ │ ├── celeba_10p.yml │ │ ├── celeba_50p.yml │ │ └── celeba_90p.yml ├── base.yml ├── celebahq_base.yml ├── celebahq_base_50p.yml ├── celebahq_base_90p.yml ├── church256_base.yml └── transfer │ ├── celebahq_small_0.01.yml │ ├── celebahq_small_0.1.yml │ ├── celebahq_small_90p.yml │ └── vggface2_small_90p.yml ├── datasets ├── __init__.py ├── base.py ├── celeba.py ├── celebahq.py ├── lsun.py ├── masking_generator.py ├── utils.py └── vggface.py ├── eval.py ├── gaussian_ddpm.py ├── main.py ├── models ├── __init__.py └── mask_uvit.py ├── tools ├── inception.py └── pytorch_fid.py ├── train_step.py └── utils ├── config.py ├── helper.py └── lr_schedule.py /.gitignore: -------------------------------------------------------------------------------- 1 | **/__pycache__ 2 | 3 | **/local** 4 | _** -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/README.md -------------------------------------------------------------------------------- /configs/.gitignore: -------------------------------------------------------------------------------- 1 | temp** 2 | local** -------------------------------------------------------------------------------- /configs/ablation/celeba_small.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/configs/ablation/celeba_small.yml -------------------------------------------------------------------------------- /configs/ablation/pretrain/block2/celeba_10p.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/configs/ablation/pretrain/block2/celeba_10p.yml -------------------------------------------------------------------------------- /configs/ablation/pretrain/block2/celeba_50p.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/configs/ablation/pretrain/block2/celeba_50p.yml -------------------------------------------------------------------------------- /configs/ablation/pretrain/block2/celeba_90p.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/configs/ablation/pretrain/block2/celeba_90p.yml -------------------------------------------------------------------------------- /configs/ablation/pretrain/block4/celeba_10p.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/configs/ablation/pretrain/block4/celeba_10p.yml -------------------------------------------------------------------------------- /configs/ablation/pretrain/block4/celeba_50p.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/configs/ablation/pretrain/block4/celeba_50p.yml -------------------------------------------------------------------------------- /configs/ablation/pretrain/block4/celeba_90p.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/configs/ablation/pretrain/block4/celeba_90p.yml -------------------------------------------------------------------------------- /configs/ablation/pretrain/block8/celeba_50p.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/configs/ablation/pretrain/block8/celeba_50p.yml -------------------------------------------------------------------------------- /configs/ablation/pretrain/crop/celeba_10p.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/configs/ablation/pretrain/crop/celeba_10p.yml -------------------------------------------------------------------------------- /configs/ablation/pretrain/crop/celeba_50p.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/configs/ablation/pretrain/crop/celeba_50p.yml -------------------------------------------------------------------------------- /configs/ablation/pretrain/crop/celeba_90p.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/configs/ablation/pretrain/crop/celeba_90p.yml -------------------------------------------------------------------------------- /configs/ablation/pretrain/patch/celeba_10p.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/configs/ablation/pretrain/patch/celeba_10p.yml -------------------------------------------------------------------------------- /configs/ablation/pretrain/patch/celeba_50p.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/configs/ablation/pretrain/patch/celeba_50p.yml -------------------------------------------------------------------------------- /configs/ablation/pretrain/patch/celeba_90p.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/configs/ablation/pretrain/patch/celeba_90p.yml -------------------------------------------------------------------------------- /configs/base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/configs/base.yml -------------------------------------------------------------------------------- /configs/celebahq_base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/configs/celebahq_base.yml -------------------------------------------------------------------------------- /configs/celebahq_base_50p.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/configs/celebahq_base_50p.yml -------------------------------------------------------------------------------- /configs/celebahq_base_90p.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/configs/celebahq_base_90p.yml -------------------------------------------------------------------------------- /configs/church256_base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/configs/church256_base.yml -------------------------------------------------------------------------------- /configs/transfer/celebahq_small_0.01.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/configs/transfer/celebahq_small_0.01.yml -------------------------------------------------------------------------------- /configs/transfer/celebahq_small_0.1.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/configs/transfer/celebahq_small_0.1.yml -------------------------------------------------------------------------------- /configs/transfer/celebahq_small_90p.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/configs/transfer/celebahq_small_90p.yml -------------------------------------------------------------------------------- /configs/transfer/vggface2_small_90p.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/configs/transfer/vggface2_small_90p.yml -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/datasets/base.py -------------------------------------------------------------------------------- /datasets/celeba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/datasets/celeba.py -------------------------------------------------------------------------------- /datasets/celebahq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/datasets/celebahq.py -------------------------------------------------------------------------------- /datasets/lsun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/datasets/lsun.py -------------------------------------------------------------------------------- /datasets/masking_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/datasets/masking_generator.py -------------------------------------------------------------------------------- /datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/datasets/utils.py -------------------------------------------------------------------------------- /datasets/vggface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/datasets/vggface.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/eval.py -------------------------------------------------------------------------------- /gaussian_ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/gaussian_ddpm.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/main.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | from .mask_uvit import MaskedUViT 2 | -------------------------------------------------------------------------------- /models/mask_uvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/models/mask_uvit.py -------------------------------------------------------------------------------- /tools/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/tools/inception.py -------------------------------------------------------------------------------- /tools/pytorch_fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/tools/pytorch_fid.py -------------------------------------------------------------------------------- /train_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/train_step.py -------------------------------------------------------------------------------- /utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/utils/config.py -------------------------------------------------------------------------------- /utils/helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/utils/helper.py -------------------------------------------------------------------------------- /utils/lr_schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiachenlei/maskdm/HEAD/utils/lr_schedule.py --------------------------------------------------------------------------------