Hand4Whole_RELEASE 项目使用教程

Hand4Whole_RELEASE 项目使用教程

Hand4Whole_RELEASE Official PyTorch implementation of "Accurate 3D Hand Pose Estimation for Whole-Body 3D Human Mesh Estimation", CVPRW 2022 (Oral.) 项目地址: https://gitcode.com/gh_mirrors/ha/Hand4Whole_RELEASE

1. 项目目录结构及介绍

Hand4Whole_RELEASE 项目的目录结构如下：

$[ROOT]
|-- data
|-- demo
|-- main
|-- tool
|-- output
|-- common
|   |-- utils
|       |-- human_model_files
|           |-- smpl
|           |-- smplx
|           |-- mano
|           |-- flame

目录结构介绍

• data: 包含数据加载代码和软链接指向图像和注释目录。
• demo: 包含演示代码。
• main: 包含用于训练或测试网络的高级代码。
• tool: 包含 AGORA 的预处理代码和 PyTorch 模型编辑代码。
• output: 包含日志、训练模型、可视化输出和测试结果。
• common: 包含 Hand4Whole 的核心代码。
  • utils: 包含实用工具代码。
    • human_model_files: 包含 smpl、smplx、mano 和 flame 3D 模型文件。

2. 项目启动文件介绍

项目的启动文件主要位于 main 目录下，包括以下几个关键文件：

• train.py: 用于训练 Hand4Whole 网络。
• test.py: 用于测试 Hand4Whole 网络。
• demo.py: 用于运行演示代码。

启动文件介绍

• train.py:

  • 功能：用于训练 Hand4Whole 网络。
  • 使用方法：在 main 目录下运行 python train.py --gpu 0-3 --lr 1e-4 --continue。

• test.py:

  • 功能：用于测试 Hand4Whole 网络。
  • 使用方法：在 main 目录下运行 python test.py --gpu 0-3 --test_epoch 6。

• demo.py:

  • 功能：用于运行演示代码。
  • 使用方法：在 demo 目录下运行 python demo.py --gpu 0。

3. 项目配置文件介绍

项目的配置文件主要位于 main 目录下的 config.py 文件中。该文件包含了训练和测试的各种配置选项。

配置文件介绍

• config.py:
  • 功能：包含训练和测试的各种配置选项。
  • 主要配置项：
    • datasets: 指定使用的数据集。
    • lr_dec_epoch: 学习率衰减的 epoch 数。
    • end_epoch: 训练的总 epoch 数。
    • trainset_3d: 指定 3D 训练数据集。
    • trainset_2d: 指定 2D 训练数据集。
    • testset: 指定测试数据集。

配置文件示例

# config.py

# 数据集配置
datasets = ['AGORA', 'EHF', 'Human36M', 'MPII', 'MPI_INF_3DHP', 'MSCOCO', 'PW3D']

# 训练配置
lr_dec_epoch = [40, 60]
end_epoch = 70
trainset_3d = ['AGORA']
trainset_2d = []
testset = 'AGORA'

通过修改 config.py 文件中的配置项，可以灵活地调整训练和测试的参数。

Hand4Whole_RELEASE Official PyTorch implementation of "Accurate 3D Hand Pose Estimation for Whole-Body 3D Human Mesh Estimation", CVPRW 2022 (Oral.) 项目地址: https://gitcode.com/gh_mirrors/ha/Hand4Whole_RELEASE