Cat Tryoff Flux

🚀 CAT-Tryoff-Flux模型卡

CAT-Tryoff-Flux是一款先進的試穿模型。它採用了與CATVTON - FLUX相同的方法。該模型能夠從人物穿著衣物的圖像中提取並重構衣物的前視圖。

🚀 快速開始

transformer = FluxTransformer2DModel.from_pretrained(
    "xiaozaa/cat-tryoff-flux", 
    torch_dtype=torch.bfloat16
)
pipe = FluxFillPipeline.from_pretrained(
    "black-forest-labs/FLUX.1-dev",
    transformer=transformer,
    torch_dtype=torch.bfloat16
).to("cuda")

✨ 主要特性

• 可用於虛擬試穿應用，讓用戶直觀看到不同服裝穿在人身上的效果。
• 能從人物穿著衣物的圖像中提取並重構衣物前視圖。

📦 安裝指南

本模型可通過命令行界面直接使用，所需參數如下：

• 輸入人物圖像
• 人物掩碼
• 服裝圖像
• 隨機種子（可選）

💻 使用示例

基礎用法

transformer = FluxTransformer2DModel.from_pretrained(
    "xiaozaa/cat-tryoff-flux", 
    torch_dtype=torch.bfloat16
)
pipe = FluxFillPipeline.from_pretrained(
    "black-forest-labs/FLUX.1-dev",
    transformer=transformer,
    torch_dtype=torch.bfloat16
).to("cuda")

📚 詳細文檔

模型詳情

模型描述

• 開發者：X/Twitter:Black Magic An

模型來源

• 倉庫：github

訓練詳情

訓練數據

VITON - HD數據集

訓練過程

對Flux1 - dev - fill進行微調

評估

總結

BibTeX引用：

@misc{chong2024catvtonconcatenationneedvirtual,
 title={CatVTON: Concatenation Is All You Need for Virtual Try-On with Diffusion Models}, 
 author={Zheng Chong and Xiao Dong and Haoxiang Li and Shiyue Zhang and Wenqing Zhang and Xujie Zhang and Hanqing Zhao and Xiaodan Liang},
 year={2024},
 eprint={2407.15886},
 archivePrefix={arXiv},
 primaryClass={cs.CV},
 url={https://arxiv.org/abs/2407.15886}, 
}
@article{lhhuang2024iclora,
  title={In-Context LoRA for Diffusion Transformers},
  author={Huang, Lianghua and Wang, Wei and Wu, Zhi-Fan and Shi, Yupeng and Dou, Huanzhang and Liang, Chen and Feng, Yutong and Liu, Yu and Zhou, Jingren},
  journal={arXiv preprint arxiv:2410.23775},
  year={2024}
}

📄 許可證

本模型使用的許可證為cc - by - nc - 2.0。

📦 模型信息