VITON-GAN 项目使用教程

1. 项目的目录结构及介绍

VITON-GAN 项目的目录结构如下：

viton-gan/
├── viton_gan/
│   ├── data/
│   ├── models/
│   ├── utils/
│   ├── train_gmm.py
│   ├── run_gmm.py
│   ├── train_tom.py
│   ├── run_tom.py
├── .gitignore
├── LICENSE
├── README.md

目录结构介绍

• viton_gan/: 项目的主要代码目录。
  • data/: 存放数据文件，包括衣物图像、人物图像、姿态关键点等。
  • models/: 存放模型定义文件。
  • utils/: 存放工具函数和辅助代码。
  • train_gmm.py: 训练 GMM（Geometric Matching Module）的脚本。
  • run_gmm.py: 运行 GMM 的脚本。
  • train_tom.py: 训练 TOM（Try-On Module）的脚本。
  • run_tom.py: 运行 TOM 的脚本。
• .gitignore: Git 忽略文件配置。
• LICENSE: 项目许可证文件。
• README.md: 项目说明文档。

2. 项目的启动文件介绍

启动文件

• train_gmm.py: 用于训练 GMM 模型，处理衣物图像的几何匹配。
• run_gmm.py: 用于运行 GMM 模型，生成衣物图像的变形。
• train_tom.py: 用于训练 TOM 模型，处理虚拟试衣。
• run_tom.py: 用于运行 TOM 模型，生成最终的虚拟试衣图像。

使用方法

1. 训练 GMM 模型：

   python train_gmm.py
   

2. 运行 GMM 模型：

   python run_gmm.py
   

3. 训练 TOM 模型：

   python train_tom.py
   

4. 运行 TOM 模型：

   python run_tom.py
   

3. 项目的配置文件介绍

配置文件

VITON-GAN 项目没有显式的配置文件，但可以通过命令行参数或环境变量进行配置。主要的配置参数包括数据路径、模型路径、训练参数等。

示例配置

在 train_gmm.py 和 train_tom.py 中，可以通过命令行参数指定数据路径和模型路径：

python train_gmm.py --data_dir /path/to/data --model_dir /path/to/model

在 run_gmm.py 和 run_tom.py 中，同样可以通过命令行参数指定输入和输出路径：

python run_gmm.py --input_dir /path/to/input --output_dir /path/to/output

通过这些命令行参数，可以灵活地配置项目的运行环境和数据路径。