nielsr HF Staff commited on
Commit
46e4671
·
verified ·
1 Parent(s): 0501b8a

Improve model card: Add pipeline tag, update paper link, and add sample usage

Browse files

This PR enhances the model card by:

- Adding the `pipeline_tag: text-to-image` metadata for improved model discoverability on the Hugging Face Hub.
- Updating the paper link from arXiv to the official Hugging Face Papers page: [IMAGGarment: Fine-Grained Garment Generation for Controllable Fashion Design](https://huggingface.co/papers/2504.13176).
- Including a "Sample Usage" section with a Python code snippet, directly extracted from the GitHub README's "How to test" instructions, to guide users on how to run inference with the model.

Files changed (1) hide show
  1. README.md +29 -9
README.md CHANGED
@@ -1,20 +1,22 @@
1
  ---
2
- tags:
3
- - text-to-image
4
- - stable-diffusion
5
- - garment generation
6
- - multi-modality
7
- license: apache-2.0
8
  language:
9
- - en
10
  library_name: diffusers
 
 
 
 
 
 
 
11
  ---
 
12
  # IMAGGarment-1: Fine-Grained Garment Generation for Controllable Fashion Design
13
 
14
 
15
  <div align="center">
16
 
17
- [**Project Page**](https://revive234.github.io/imaggarment.github.io/) **|** [**Paper**](https://arxiv.org/pdf/2504.13176) **|** [**Code**](https://github.com/muzishen/IMAGGarment-1)
18
 
19
  </div>
20
 
@@ -24,4 +26,22 @@ library_name: diffusers
24
 
25
  IMAGGarment-1 addresses the challenges of multi-conditional controllability in personalized fashion design and digital apparel applications. Specifically, IMAGGarment-1 employs a two-stage training strategy to separately model global appearance and local details, while enabling unified and controllable generation through end-to-end inference. In the first stage, we propose a global appearance model that jointly encodes silhouette and color using a mixed attention module and a color adapter. In the second stage, we present a local enhancement model with an adaptive appearance-aware module to inject user-defined logos and spatial constraints, enabling accurate placement and visual consistency.
26
 
27
- ![framework](architecture.png)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
 
 
 
 
 
 
2
  language:
3
+ - en
4
  library_name: diffusers
5
+ license: apache-2.0
6
+ pipeline_tag: text-to-image
7
+ tags:
8
+ - text-to-image
9
+ - stable-diffusion
10
+ - garment generation
11
+ - multi-modality
12
  ---
13
+
14
  # IMAGGarment-1: Fine-Grained Garment Generation for Controllable Fashion Design
15
 
16
 
17
  <div align="center">
18
 
19
+ [**Project Page**](https://revive234.github.io/imaggarment.github.io/) **|** [**Paper**](https://huggingface.co/papers/2504.13176) **|** [**Code**](https://github.com/muzishen/IMAGGarment-1)
20
 
21
  </div>
22
 
 
26
 
27
  IMAGGarment-1 addresses the challenges of multi-conditional controllability in personalized fashion design and digital apparel applications. Specifically, IMAGGarment-1 employs a two-stage training strategy to separately model global appearance and local details, while enabling unified and controllable generation through end-to-end inference. In the first stage, we propose a global appearance model that jointly encodes silhouette and color using a mixed attention module and a color adapter. In the second stage, we present a local enhancement model with an adaptive appearance-aware module to inject user-defined logos and spatial constraints, enabling accurate placement and visual consistency.
28
 
29
+ ![framework](architecture.png)
30
+
31
+ ## Sample Usage
32
+
33
+ To test the model, you can use the following inference code as demonstrated in the GitHub repository:
34
+
35
+ ```python
36
+ python inference_IMAGGarment-1.py \
37
+ --GAM_model_ckpt [GAM checkpoint] \
38
+ --LEM_model_ckpt [LEM chekcpoint] \
39
+ --sketch_path [your sketch path] \
40
+ --logo_path [your logo path] \
41
+ --mask_path [your mask path] \
42
+ --color_path [your color path] \
43
+ --prompt [your prompt] \
44
+ --output_path [your save path] \
45
+ --color_ckpt [color adapter checkpoint] \
46
+ --device [your device]
47
+ ```