Update README.md
Browse files
README.md
CHANGED
@@ -19,14 +19,14 @@ library_name: diffusers
|
|
19 |
|
20 |
These models are based on the research papers:
|
21 |
1. _"TryOffDiff: Virtual-Try-Off via High-Fidelity Garment Reconstruction using Diffusion Models"_ [[paper]][paper_arxiv]
|
22 |
-
2. _"
|
23 |
|
24 |
Available models:
|
25 |
- `tryoffdiff.pth`: Generates upper-body garments. Fine-tuned on `VITON-HD-train`.
|
26 |
- `tryoffdiffv2_upper.pth`: Generates upper-body garments. Fine-tuned on a subset of `Dress Code-train`.
|
27 |
- `tryoffdiffv2_lower.pth`: Generates lower-body garments. Fine-tuned on a subset of `Dress Code-train`.
|
28 |
- `tryoffdiffv2_dress.pth`: Generates dresses. Fine-tuned on a subset of `Dress Code-train`.
|
29 |
-
- `tryoffdiffv2_multi.pth
|
30 |
|
31 |
## Usage
|
32 |
|
@@ -49,10 +49,10 @@ If you use this work, please give a star ⭐ and a citation:
|
|
49 |
year = {2024},
|
50 |
note = {\url{https://doi.org/nt3n}}
|
51 |
}
|
52 |
-
@
|
53 |
-
title = {
|
54 |
author = {Velioglu, Riza and Bevandic, Petra and Chan, Robin and Hammer, Barbara},
|
55 |
-
|
56 |
year = {2025},
|
57 |
note = {\url{https://doi.org/pn67}}
|
58 |
}
|
|
|
19 |
|
20 |
These models are based on the research papers:
|
21 |
1. _"TryOffDiff: Virtual-Try-Off via High-Fidelity Garment Reconstruction using Diffusion Models"_ [[paper]][paper_arxiv]
|
22 |
+
2. _"MGT: Extending Virtual Try-Off to Multi-Garment Scenarios"_ [[paper]][paper2_arxiv]
|
23 |
|
24 |
Available models:
|
25 |
- `tryoffdiff.pth`: Generates upper-body garments. Fine-tuned on `VITON-HD-train`.
|
26 |
- `tryoffdiffv2_upper.pth`: Generates upper-body garments. Fine-tuned on a subset of `Dress Code-train`.
|
27 |
- `tryoffdiffv2_lower.pth`: Generates lower-body garments. Fine-tuned on a subset of `Dress Code-train`.
|
28 |
- `tryoffdiffv2_dress.pth`: Generates dresses. Fine-tuned on a subset of `Dress Code-train`.
|
29 |
+
- `tryoffdiffv2_multi.pth` (*MGT*): Generates all three garment types (upper-body, lower-body, dresses). Fine-tuned on the entire `Dress Code-train`.
|
30 |
|
31 |
## Usage
|
32 |
|
|
|
49 |
year = {2024},
|
50 |
note = {\url{https://doi.org/nt3n}}
|
51 |
}
|
52 |
+
@inproceedings{velioglu2025mgt,
|
53 |
+
title = {MGT: Extending Virtual Try-Off to Multi-Garment Scenarios},
|
54 |
author = {Velioglu, Riza and Bevandic, Petra and Chan, Robin and Hammer, Barbara},
|
55 |
+
booktitle = {ICCVW},
|
56 |
year = {2025},
|
57 |
note = {\url{https://doi.org/pn67}}
|
58 |
}
|