DirectMHP 项目使用教程

DirectMHP 项目使用教程

DirectMHP Codes for my paper "DirectMHP: Direct 2D Multi-Person Head Pose Estimation with Full-range Angles" 项目地址: https://gitcode.com/gh_mirrors/di/DirectMHP

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

DirectMHP/
├── Sim3DR/
│   ├── build_sim3dr.sh
│   └── ...
├── data/
│   ├── 300w_lp_aflw2000.yaml
│   ├── 300w_lp_biwi.yaml
│   └── ...
├── demos/
│   ├── demo_script.py
│   └── ...
├── exps/
│   ├── convert_coco_style_300wlp_aflw2000.py
│   ├── convert_coco_style_300wlp_biwi.py
│   └── ...
├── materials/
│   ├── material_file.txt
│   └── ...
├── models/
│   ├── model_config.json
│   └── ...
├── pose_references/
│   ├── pose_reference_file.txt
│   └── ...
├── test_imgs/
│   ├── test_image1.jpg
│   └── ...
├── utils/
│   ├── labels.py
│   └── ...
├── weights/
│   ├── weight_file.pt
│   └── ...
├── LICENSE.txt
├── README.md
├── requirements.txt
├── train.py
└── val.py

目录结构介绍

• Sim3DR/: 包含用于可视化预测结果的渲染器相关文件。
• data/: 包含项目的配置文件，如数据集配置文件。
• demos/: 包含演示脚本，用于展示项目的功能。
• exps/: 包含实验相关的脚本，如数据转换脚本。
• materials/: 包含项目所需的各种材料文件。
• models/: 包含模型配置文件。
• pose_references/: 包含姿态参考文件。
• test_imgs/: 包含测试图像。
• utils/: 包含各种实用工具脚本。
• weights/: 包含预训练的权重文件。
• LICENSE.txt: 项目的许可证文件。
• README.md: 项目的说明文档。
• requirements.txt: 项目依赖的Python包列表。
• train.py: 项目的训练脚本。
• val.py: 项目的验证脚本。

2. 项目的启动文件介绍

train.py

train.py 是项目的训练脚本，用于训练模型。启动训练的命令如下：

python train.py --config config/train_config.yaml

val.py

val.py 是项目的验证脚本，用于验证模型的性能。启动验证的命令如下：

python val.py --config config/val_config.yaml

3. 项目的配置文件介绍

data/300w_lp_aflw2000.yaml

该配置文件用于配置300W-LP和AFLW2000数据集的训练和验证设置。

data/300w_lp_biwi.yaml

该配置文件用于配置300W-LP和BIWI数据集的训练和验证设置。

config/train_config.yaml

该配置文件包含训练过程中的各种参数设置，如学习率、批量大小、训练轮数等。

config/val_config.yaml

该配置文件包含验证过程中的各种参数设置，如验证数据集路径、验证批量大小等。

通过这些配置文件，用户可以灵活地调整项目的运行参数，以适应不同的训练和验证需求。

DirectMHP Codes for my paper "DirectMHP: Direct 2D Multi-Person Head Pose Estimation with Full-range Angles" 项目地址: https://gitcode.com/gh_mirrors/di/DirectMHP