TM2T 开源项目使用教程

TM2T 开源项目使用教程

TM2TOfficial implementation of "TM2T: Stochastic and Tokenized Modeling for the Reciprocal Generation of 3D Human Motions and Texts (ECCV2022)"项目地址:https://gitcode.com/gh_mirrors/tm/TM2T

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

TM2T/
├── data/
│   ├── HumanML3D/
│   └── KIT-ML/
├── models/
│   ├── motion_model.py
│   └── text_model.py
├── scripts/
│   ├── train.py
│   ├── evaluate.py
│   └── generate.py
├── configs/
│   ├── default_config.yaml
│   └── custom_config.yaml
├── README.md
└── requirements.txt

• data/: 包含项目所需的数据集，如 HumanML3D 和 KIT-ML。
• models/: 包含项目的模型文件，如 motion_model.py 和 text_model.py。
• scripts/: 包含项目的脚本文件，如训练脚本 train.py、评估脚本 evaluate.py 和生成脚本 generate.py。
• configs/: 包含项目的配置文件，如 default_config.yaml 和 custom_config.yaml。
• README.md: 项目的说明文档。
• requirements.txt: 项目所需的依赖包列表。

2. 项目的启动文件介绍

项目的启动文件主要位于 scripts/ 目录下：

• train.py: 用于训练模型的脚本。
• evaluate.py: 用于评估模型性能的脚本。
• generate.py: 用于生成 3D 动画的脚本。

示例命令

# 训练模型
python scripts/train.py --config configs/default_config.yaml

# 评估模型
python scripts/evaluate.py --config configs/default_config.yaml

# 生成 3D 动画
python scripts/generate.py --config configs/default_config.yaml

3. 项目的配置文件介绍

项目的配置文件位于 configs/ 目录下：

• default_config.yaml: 默认配置文件，包含模型训练、评估和生成的默认参数。
• custom_config.yaml: 自定义配置文件，用户可以根据需要修改参数。

配置文件示例

# default_config.yaml
train:
  batch_size: 32
  learning_rate: 0.001
  epochs: 100

evaluate:
  dataset: HumanML3D
  metrics: [accuracy, f1_score]

generate:
  output_format: mp4
  sample_rate: 30

用户可以根据自己的需求修改配置文件中的参数，以适应不同的训练、评估和生成任务。

TM2TOfficial implementation of "TM2T: Stochastic and Tokenized Modeling for the Reciprocal Generation of 3D Human Motions and Texts (ECCV2022)"项目地址:https://gitcode.com/gh_mirrors/tm/TM2T