CORE4D
收藏CORE4D: A 4D Human-Object-Human Interaction Dataset for Collaborative Object REarrangement
数据组织
数据集的组织结构如下:
|--CORE4D_Real |--object_models ... |--human_object_motions ... |--allocentric_RGBD_videos ... |--egocentric_RGB_videos ... |--human_object_segmentations ... |--camera_parameters ... |--action_labels.json |--CORE4D_Synthetic |-- <motion sequence name 1> |-- human_poses.npy |-- object_mesh.obj |-- object_poses.npy |-- <motion sequence name 2> |-- human_poses.npy |-- object_mesh.obj |-- object_poses.npy ...
文件定义
详细的文件定义请参考 docs/file_definitions.md。
数据可视化
环境设置
代码测试环境为 Ubuntu 20.04,配备 NVIDIA GeForce RTX 3090 GPU,驱动版本为 535.129.03,CUDA 版本为 12.2。
设置环境的命令如下:
x conda create -n core4d python=3.9 conda activate core4d <install PyTorch >= 1.7.1> <install PyTorch3D >= 0.6.1> cd dataset_utils pip install -r requirements.txt
可视化人体-物体运动
可视化人体-物体运动的命令如下:
x cd dataset_utils python visualize_human_object_motion.py --dataset_root <dataset root directory> --object_model_root <object model root directory> --sequence_name <sequence name> --save_path <path to save the visualization result> --device <device for the rendering process>
例如,选择以下数据序列:
x python visualize_human_object_motion.py --dataset_root <dataset root directory> --object_model_root <object model root directory> --sequence_name "20231002/004" --save_path "./example.gif" --device "cuda:0"
可视化结果示例:
<img src="https://raw.githubusercontent.com/leolyliu/CORE4D-Instructions/main/assets/example.gif" width="1920"/>
基准代码
人体-物体运动预测
实现基准 "human-object motion forecasting" 请参考 ./benchmarks/motion_forecasting/README.md。
交互合成
实现基准 "interaction synthesis" 请参考 ./benchmarks/interaction_synthesis/README.md。
许可证
本工作采用 CC BY 4.0 许可证。
引用
如果本工作对您有帮助,请引用:
x @article{zhang2024core4d, title={CORE4D: A 4D Human-Object-Human Interaction Dataset for Collaborative Object REarrangement}, author={Zhang, Chengwen and Liu, Yun and Xing, Ruofan and Tang, Bingda and Yi, Li}, journal={arXiv preprint arXiv:2406.19353}, year={2024} }

- 1CORE4D: A 4D Human-Object-Human Interaction Dataset for Collaborative Object REarrangement清华大学交叉信息研究院, 北京邮电大学, 上海人工智能实验室, 上海期智研究院 · 2024年



