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
用户可以根据自己的需求修改配置文件中的参数,以适应不同的训练、评估和生成任务。