|
|
--- |
|
|
license: apache-2.0 |
|
|
language: |
|
|
- en |
|
|
- zh |
|
|
--- |
|
|
|
|
|
# GarmentGPT Models |
|
|
|
|
|
This repository contains all the necessary model components for the GarmentGPT project. |
|
|
|
|
|
## Models Included |
|
|
|
|
|
This repository hosts three key components: |
|
|
|
|
|
1. **Vision-Language Model (LLM)**: A fine-tuned multi-modal model responsible for generating discrete garment tokens from an input image. |
|
|
2. **Edge Codec**: A VQ-VAE-based model for decoding edge indices into high-fidelity geometric curves. The configuration is in `codec_config.yaml` and weights are in `codec_model.pth`. |
|
|
3. **RT Codec**: A VQ-VAE-based model for decoding location indices into 3D panel rotation and translation. The configuration is in `rt_config.yaml` and weights are in `rt_model.pth`. |
|
|
|
|
|
## Usage |
|
|
|
|
|
These models are designed to be used with the main application code available at https://github.com/ChimerAI-MMLab/Garment-GPT. The inference script will automatically download these files. |
|
|
|
