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 |
}
|