D3DP 项目使用教程
D3DP项目地址:https://gitcode.com/gh_mirrors/d3/d3dp
1. 项目的目录结构及介绍
D3DP 项目的目录结构如下:
d3dp/
├── checkpoint/
├── common/
├── data/
├── fig/
├── in_the_wild/
├── 3dhp_test/
├── LICENSE
├── README.md
├── main.py
├── main_3dhp.py
├── main_draw.py
├── main_in_the_wild.py
各目录和文件的介绍如下:
checkpoint/
:存放预训练模型的目录。common/
:包含项目通用代码的目录。data/
:存放数据集的目录。fig/
:存放生成的图表的目录。in_the_wild/
:存放野外视频测试相关代码的目录。3dhp_test/
:存放 MPI-INF-3DHP 数据集测试相关代码的目录。LICENSE
:项目的许可证文件。README.md
:项目的说明文档。main.py
:项目的主启动文件。main_3dhp.py
:用于 MPI-INF-3DHP 数据集的启动文件。main_draw.py
:用于可视化的启动文件。main_in_the_wild.py
:用于野外视频测试的启动文件。
2. 项目的启动文件介绍
main.py
这是项目的主启动文件,用于训练和评估模型。主要参数包括:
-k
:指定使用的 2D 关键点检测器。-c
:指定检查点目录。-gpu
:指定使用的 GPU 设备。--nolog
:禁用日志记录。--evaluate
:评估预训练模型。
示例命令:
python main.py -k cpn_ft_h36m_dbb -c checkpoint -gpu 0 --nolog --evaluate h36m_best_epoch.bin
main_3dhp.py
用于 MPI-INF-3DHP 数据集的启动文件。主要参数与 main.py
类似,但针对 MPI-INF-3DHP 数据集进行了优化。
示例命令:
python main_3dhp.py -c checkpoint -gpu 0 --nolog --evaluate 3dhp_best_epoch.bin
main_draw.py
用于可视化的启动文件。可以生成可视化结果并保存到 fig/
目录。
示例命令:
python main_draw.py -k cpn_ft_h36m_dbb -b 2 -c checkpoint -gpu 0 --nolog --evaluate h36m_best_epoch.bin --render --viz-subject S11 --viz-action SittingDown --viz-camera 1
main_in_the_wild.py
用于野外视频测试的启动文件。可以处理自定义视频并生成 3D 姿态。
示例命令:
python main_in_the_wild.py -c checkpoint/model_3dhp -gpu 0 --nolog
3. 项目的配置文件介绍
项目没有显式的配置文件,但可以通过命令行参数进行配置。主要的配置参数包括:
-k
:指定使用的 2D 关键点检测器。-c
:指定检查点目录。-gpu
:指定使用的 GPU 设备。--nolog
:禁用日志记录。--evaluate
:评估预训练模型。
这些参数可以在启动文件的命令行中进行设置,以满足不同的训练和评估需求。
以上是 D3DP 项目的基本使用教程,涵盖了项目的目录结构、启动文件和配置方法。希望这些信息能帮助你更好地理解和使用该项目。