TH ¨OR-MAGNI Act
收藏THÖR-MAGNI Act 数据集概述
数据集简介
THÖR-MAGNI Act 数据集是一个用于人机共享工业空间中人体运动建模的动作数据集。该数据集提供了用于建模的原始数据、中间数据和最终处理后的数据集。
数据结构
数据集的组织结构如下:
- data/raw: 原始数据,不可变的数据集。
- data/external: 来自第三方的数据。
- data/interim: 经过转换的中间数据。
- data/processed: 最终用于建模的规范数据集。
数据预处理步骤
-
准备
thor-magni-tools工具。 -
修改配置文件
cfg.yaml,设置输入路径 (in_path) 和输出路径 (out_path)。 -
从
thor-magni-tools运行预处理脚本:python -m thor_magni_tools.run_preprocessing
-
检查
data/external目录。 -
对每个预处理的场景目录,运行以下命令以对齐动作和轨迹数据:
unzip data/processed/thor_magni/QTM_frames_actions.zip -d data/processed/thor_magni/ && python -m thor_magni_tools.run_actions_merging --actions_path data/processed/thor_magni/QTM_frames_actions.csv --files_dir outputs/data/thor_magni/Scenario_{ID}/ --out_path data/interim/thor_magni/
-
计算特征:
python -m thor_magni_actions.features.build_features data/interim/thor_magni data/interim/thor_magni
-
创建固定长度的轨迹片段数据集:
python -m thor_magni_actions.data.make_dataset thor_magni data/interim/thor_magni data/processed/thor_magni
模型训练与验证
运行 k-fold 交叉验证和多任务学习方法:
python -m thor_magni_actions.data_modeling.runners.k_fold_cv 5 thor_magni_actions/data_modeling/cfgs/mtl_tf.yaml

- 1THÖR-MAGNI Act: Actions for Human Motion Modeling in Robot-Shared Industrial Spaces厄勒布鲁大学 · 2024年



