|
---
|
|
license: other
|
|
license_name: server-side-public-license
|
|
license_link: https://www.mongodb.com/legal/licensing/server-side-public-license
|
|
tags:
|
|
- diffusion
|
|
- virtual try-on
|
|
- virtual try-off
|
|
- image generation
|
|
- fashion
|
|
- e-commerce
|
|
base_model:
|
|
- CompVis/stable-diffusion-v1-4
|
|
pipeline_tag: image-to-image
|
|
library_name: diffusers
|
|
---
|
|
|
|
## TryOffDiff
|
|
|
|
The models proposed in the paper _"TryOffDiff: Virtual-Try-Off via High-Fidelity Garment Reconstruction\\using Diffusion Models"_
|
|
[[paper]](https://arxiv.org/abs/2404.08582) [[project page]][project_page]:
|
|
- `tryoffdiff.pth`: The pre-trained StableDiffusion-v1.4 fine-tuned on `VITON-HD-train` dataset.
|
|
- `.pth`: A U-Net trained from scratch on `VITON-HD-train` dataset.
|
|
- `.pth`:
|
|
|
|
|
|
## Usage
|
|
|
|
```python
|
|
from torchvision.io import read_image
|
|
from torchvision.models.detection import MaskRCNN_ResNet50_FPN_Weights
|
|
from huggingface_hub import hf_hub_download
|
|
|
|
path_onnx = hf_hub_download(
|
|
repo_id="rizavelioglu/tryoffdiff",
|
|
filename="tryoffdiff.pth", # or one of ["ldm-1", "ldm-2", "ldm-3", ...]
|
|
)
|
|
|
|
```
|
|
|
|
> Check out the demo code on [HuggingFace Spaces][hf_spaces] for visualizing the output.
|
|
|
|
> Also, check out [GitHub repository][project_page] to get more information on
|
|
> training, inference, and evaluation.
|
|
|
|
### License
|
|
TL;DR: Not available for commercial use, unless the FULL source code is shared! \
|
|
This project is intended solely for academic research. No commercial benefits are derived from it.
|
|
Models are licensed under [Server Side Public License (SSPL)][license]
|
|
|
|
### Citation
|
|
If you find this repository useful in your research, please consider giving a star ⭐ and a citation:
|
|
```
|
|
@article{velioglu2024tryoffdiff,
|
|
title = {TryOffDiff: Virtual-Try-Off via High-Fidelity Garment Reconstruction using Diffusion Models},
|
|
author = {Velioglu, Riza and Bevandic, Petra and Chan, Robin and Hammer, Barbara},
|
|
journal = {arXiv},
|
|
year = {2024},
|
|
note = {\url{https://doi.org/....}}
|
|
}
|
|
```
|
|
|
|
[hf_spaces]: https://huggingface.co./spaces/rizavelioglu/tryoffdiff
|
|
[project_page]: https://rizavelioglu.github.io/tryoffdiff/
|
|
[github]: https://github.com/rizavelioglu/tryoffdiff
|
|
[license]: https://www.mongodb.com/legal/licensing/server-side-public-license
|
|
|
|
|
