aiozai/AIOZ-GDANCE
收藏AIOZ-GDANCE 数据集
概述
AIOZ-GDANCE 数据集是一个用于音乐驱动群舞生成的大型数据集,包含16.7小时的配对音乐和3D舞蹈动作,涵盖7种舞蹈风格和16种音乐类型。该数据集支持群舞研究,并提供了一种半自主的标注方法,通过人在环路中获取3D地面真实数据。
数据集下载
数据结构
数据集目录结构如下:
- split_sequence_names.txt: 包含数据中各个序列名称的txt文件。
- musics: 包含每个序列对应的原始音乐.wav文件,音乐帧与动作帧对齐。
- motions_smpl: 包含每个序列对应的.pkl格式动作文件,主要包含以下项目:
smpl_poses: shape[num_persons x num_frames x 72]: 72维向量姿态序列,采用SMPL姿态格式(24个关节)。root_trans: shape[num_persons x num_frames x 3]: 根翻译序列。
示例代码
以下是一个读取动作文件的Python示例代码: python import pickle import numpy as np data = pickle.load(open("sequence_name.pkl","rb")) print(data.keys())
smpl_poses = data[smpl_poses] smpl_trans = data[root_trans]
可视化
数据集提供了加载和可视化动作的演示代码。
前提条件
需要下载SMPL模型(v1.0.0)并重命名模型文件。数据目录结构应为:
<DATA_DIR> ├── motions_smpl/ ├── musics/ └── split_sequence_names.txt
<SMPL_DIR> ├── SMPL_MALE.pkl └── SMPL_FEMALE.pkl
安装必要的包:
pip install scipy torch smplx chumpy vedo trimesh pip install numpy==1.23.0
使用方法
可视化SMPL关节
以下命令将计算SMPL关节位置并在3D图形中实时绘制: bash python vis_smpl_kpt.py --data_dir <DATA_DIR>/motions_smpl --smpl_path <SMPL_DIR>/SMPL_FEMALE.PKL --sequence_name sequence_name.pkl
可视化SMPL网格
以下命令将计算SMPL网格并在3D中可视化: bash python vis_smpl_mesh.py --data_dir <DATA_DIR>/motions_smpl --smpl_path <SMPL_DIR>/SMPL_FEMALE.PKL --sequence_name sequence_name.pkl
引用
@inproceedings{aiozGdance, author = {Le, Nhat and Pham, Thang and Do, Tuong and Tjiputra, Erman and Tran, Quang D. and Nguyen, Anh}, title = {Music-Driven Group Choreography}, journal = {CVPR}, year = {2023}, }
许可证
软件版权许可用于非商业科学研究目的。请仔细阅读以下条款和条件。




