├── .gitignore ├── README.md ├── app.py ├── assets ├── images │ ├── garment │ │ ├── 00126_00.jpg │ │ ├── 04743_00.jpg │ │ └── 13987_00.jpg │ ├── image_parse │ │ ├── 01163_00_agn.jpg │ │ ├── 01163_00_densepose.png │ │ ├── 01163_00_mask.png │ │ ├── 01827_00_agn.jpg │ │ ├── 01827_00_densepose.png │ │ ├── 01827_00_mask.png │ │ ├── 13987_00_agn.jpg │ │ ├── 13987_00_densepose.png │ │ └── 13987_00_mask.png │ └── model │ │ ├── 01163_00.jpg │ │ ├── 01827_00.jpg │ │ └── 13987_00.jpg └── teaser.jpg ├── infer_t2i.py ├── infer_tryon.py ├── requirements.txt ├── setup.py ├── stablegarment ├── __init__.py ├── data │ ├── dresscode.py │ ├── generate_mask.py │ ├── labelmap.py │ ├── pack.py │ ├── posemap.py │ ├── synthesis.py │ ├── utils.py │ └── vitonhd.py ├── models │ ├── __init__.py │ ├── controlnet.py │ ├── controlnet_lora.py │ ├── garment_encoder.py │ ├── self_attention_modules.py │ ├── self_attention_modules_csa.py │ └── utils.py ├── pipelines │ ├── __init__.py │ ├── pipeline_controlnet_tryon_attn.py │ ├── pipeline_controlnet_tryon_attn_sdxl.py │ ├── pipeline_text2img_attn.py │ └── pipeline_text2img_attn_sdxl.py └── utils │ ├── save_controlnet_config.py │ └── util.py └── test.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/README.md -------------------------------------------------------------------------------- /app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/app.py -------------------------------------------------------------------------------- /assets/images/garment/00126_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/assets/images/garment/00126_00.jpg -------------------------------------------------------------------------------- /assets/images/garment/04743_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/assets/images/garment/04743_00.jpg -------------------------------------------------------------------------------- /assets/images/garment/13987_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/assets/images/garment/13987_00.jpg -------------------------------------------------------------------------------- /assets/images/image_parse/01163_00_agn.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/assets/images/image_parse/01163_00_agn.jpg -------------------------------------------------------------------------------- /assets/images/image_parse/01163_00_densepose.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/assets/images/image_parse/01163_00_densepose.png -------------------------------------------------------------------------------- /assets/images/image_parse/01163_00_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/assets/images/image_parse/01163_00_mask.png -------------------------------------------------------------------------------- /assets/images/image_parse/01827_00_agn.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/assets/images/image_parse/01827_00_agn.jpg -------------------------------------------------------------------------------- /assets/images/image_parse/01827_00_densepose.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/assets/images/image_parse/01827_00_densepose.png -------------------------------------------------------------------------------- /assets/images/image_parse/01827_00_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/assets/images/image_parse/01827_00_mask.png -------------------------------------------------------------------------------- /assets/images/image_parse/13987_00_agn.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/assets/images/image_parse/13987_00_agn.jpg -------------------------------------------------------------------------------- /assets/images/image_parse/13987_00_densepose.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/assets/images/image_parse/13987_00_densepose.png -------------------------------------------------------------------------------- /assets/images/image_parse/13987_00_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/assets/images/image_parse/13987_00_mask.png -------------------------------------------------------------------------------- /assets/images/model/01163_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/assets/images/model/01163_00.jpg -------------------------------------------------------------------------------- /assets/images/model/01827_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/assets/images/model/01827_00.jpg -------------------------------------------------------------------------------- /assets/images/model/13987_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/assets/images/model/13987_00.jpg -------------------------------------------------------------------------------- /assets/teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/assets/teaser.jpg -------------------------------------------------------------------------------- /infer_t2i.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/infer_t2i.py -------------------------------------------------------------------------------- /infer_tryon.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/infer_tryon.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/setup.py -------------------------------------------------------------------------------- /stablegarment/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/stablegarment/__init__.py -------------------------------------------------------------------------------- /stablegarment/data/dresscode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/stablegarment/data/dresscode.py -------------------------------------------------------------------------------- /stablegarment/data/generate_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/stablegarment/data/generate_mask.py -------------------------------------------------------------------------------- /stablegarment/data/labelmap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/stablegarment/data/labelmap.py -------------------------------------------------------------------------------- /stablegarment/data/pack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/stablegarment/data/pack.py -------------------------------------------------------------------------------- /stablegarment/data/posemap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/stablegarment/data/posemap.py -------------------------------------------------------------------------------- /stablegarment/data/synthesis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/stablegarment/data/synthesis.py -------------------------------------------------------------------------------- /stablegarment/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/stablegarment/data/utils.py -------------------------------------------------------------------------------- /stablegarment/data/vitonhd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/stablegarment/data/vitonhd.py -------------------------------------------------------------------------------- /stablegarment/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/stablegarment/models/__init__.py -------------------------------------------------------------------------------- /stablegarment/models/controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/stablegarment/models/controlnet.py -------------------------------------------------------------------------------- /stablegarment/models/controlnet_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/stablegarment/models/controlnet_lora.py -------------------------------------------------------------------------------- /stablegarment/models/garment_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/stablegarment/models/garment_encoder.py -------------------------------------------------------------------------------- /stablegarment/models/self_attention_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/stablegarment/models/self_attention_modules.py -------------------------------------------------------------------------------- /stablegarment/models/self_attention_modules_csa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/stablegarment/models/self_attention_modules_csa.py -------------------------------------------------------------------------------- /stablegarment/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/stablegarment/models/utils.py -------------------------------------------------------------------------------- /stablegarment/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/stablegarment/pipelines/__init__.py -------------------------------------------------------------------------------- /stablegarment/pipelines/pipeline_controlnet_tryon_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/stablegarment/pipelines/pipeline_controlnet_tryon_attn.py -------------------------------------------------------------------------------- /stablegarment/pipelines/pipeline_controlnet_tryon_attn_sdxl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/stablegarment/pipelines/pipeline_controlnet_tryon_attn_sdxl.py -------------------------------------------------------------------------------- /stablegarment/pipelines/pipeline_text2img_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/stablegarment/pipelines/pipeline_text2img_attn.py -------------------------------------------------------------------------------- /stablegarment/pipelines/pipeline_text2img_attn_sdxl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/stablegarment/pipelines/pipeline_text2img_attn_sdxl.py -------------------------------------------------------------------------------- /stablegarment/utils/save_controlnet_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/stablegarment/utils/save_controlnet_config.py -------------------------------------------------------------------------------- /stablegarment/utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/stablegarment/utils/util.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/logn-2024/StableGarment/HEAD/test.py --------------------------------------------------------------------------------