├── DATA ├── DressCode └── zalando-hd-resized ├── LICENSE.txt ├── README.md ├── environment.sh ├── eval.sh ├── generate_dc.py ├── generate_dc.sh ├── generate_vitonhd.py ├── generate_vitonhd.sh ├── image_train.py ├── images_github └── teaser.png ├── masked_diffusion ├── __init__.py ├── diffusion_utils.py ├── dist_util.py ├── fp16_util.py ├── gaussian_diffusion.py ├── logger.py ├── models.py ├── nn.py ├── resample.py ├── respace.py ├── script_util.py ├── timestep_sampler.py └── train_util.py ├── preprocess_salient_region_extraction.py ├── scr ├── dataset_dresscode.py ├── dataset_vitonhd.py ├── eval.py └── generate_fid_stats.py ├── setup.py └── train.sh /DATA/DressCode: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /DATA/zalando-hd-resized: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/README.md -------------------------------------------------------------------------------- /environment.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/environment.sh -------------------------------------------------------------------------------- /eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/eval.sh -------------------------------------------------------------------------------- /generate_dc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/generate_dc.py -------------------------------------------------------------------------------- /generate_dc.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/generate_dc.sh -------------------------------------------------------------------------------- /generate_vitonhd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/generate_vitonhd.py -------------------------------------------------------------------------------- /generate_vitonhd.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/generate_vitonhd.sh -------------------------------------------------------------------------------- /image_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/image_train.py -------------------------------------------------------------------------------- /images_github/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/images_github/teaser.png -------------------------------------------------------------------------------- /masked_diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/masked_diffusion/__init__.py -------------------------------------------------------------------------------- /masked_diffusion/diffusion_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/masked_diffusion/diffusion_utils.py -------------------------------------------------------------------------------- /masked_diffusion/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/masked_diffusion/dist_util.py -------------------------------------------------------------------------------- /masked_diffusion/fp16_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/masked_diffusion/fp16_util.py -------------------------------------------------------------------------------- /masked_diffusion/gaussian_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/masked_diffusion/gaussian_diffusion.py -------------------------------------------------------------------------------- /masked_diffusion/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/masked_diffusion/logger.py -------------------------------------------------------------------------------- /masked_diffusion/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/masked_diffusion/models.py -------------------------------------------------------------------------------- /masked_diffusion/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/masked_diffusion/nn.py -------------------------------------------------------------------------------- /masked_diffusion/resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/masked_diffusion/resample.py -------------------------------------------------------------------------------- /masked_diffusion/respace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/masked_diffusion/respace.py -------------------------------------------------------------------------------- /masked_diffusion/script_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/masked_diffusion/script_util.py -------------------------------------------------------------------------------- /masked_diffusion/timestep_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/masked_diffusion/timestep_sampler.py -------------------------------------------------------------------------------- /masked_diffusion/train_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/masked_diffusion/train_util.py -------------------------------------------------------------------------------- /preprocess_salient_region_extraction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/preprocess_salient_region_extraction.py -------------------------------------------------------------------------------- /scr/dataset_dresscode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/scr/dataset_dresscode.py -------------------------------------------------------------------------------- /scr/dataset_vitonhd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/scr/dataset_vitonhd.py -------------------------------------------------------------------------------- /scr/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/scr/eval.py -------------------------------------------------------------------------------- /scr/generate_fid_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/scr/generate_fid_stats.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/setup.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiwoohong93/ita-mdt_code/HEAD/train.sh --------------------------------------------------------------------------------