ARTICLE · INTELLIGENCE

战地情报 · 详情页

来自尧图项目组的一线实战观察与深度解析

630小时动捕数据如何炼成?ARDY-SOMA-RP-30FPS-KimodoRep-Horizon60动作生成模型训练数据、LLM增强与评估指标解析

630小时动捕数据如何炼成?ARDY-SOMA-RP-30FPS-KimodoRep-Horizon60动作生成模型训练数据、LLM增强与评估指标解析
  • 人工智能
  • 媒体生成
  • 数字人
  • 多模态

【免费下载链接】ARDY-SOMA-RP-30FPS-KimodoRep-Horizon60

项目地址:https://ai.gitcode.com/hf_mirrors/nvidia/ARDY-SOMA-RP-30FPS-KimodoRep-Horizon60
点击查看免费下载

ARDY-SOMA-RP-30FPS-KimodoRep-Horizon60 是 NVIDIA 发布的人类动作生成模型,基于 630 小时光学动捕数据训练,并借助 LLM 对文本标签做多样化增强,可实时生成交互式人形角色动作。本文带你完整拆解它的训练数据流水线、LLM 数据增强方案与评估指标体系。

🎯 一句话看懂:ARDY 是什么?

ARDY(Autoregressive Diffusion with Hybrid Representation)是一个为交互式动作生成设计的自回归扩散模型,支持在线文本提示与灵活长时程运动学约束(根路径/路径点、全身关键帧、稀疏关节位置/旋转),具备实时响应能力。

本仓库对应的模型变体由 4 个关键标识组成:

标识含义
SOMA-RP基于 30 关节 SOMA 人体模型骨架,使用 Bones Rigplay 1 动捕数据训练
30FPS动作帧率为 30 fps,训练时动作统一重采样到该帧率
Horizon60单次生成的时间窗口(horizon)为 60 帧,即 30 fps 下约 2 秒动作
KimodoRep采用 Kimodo 模型使用的平滑根运动表示(smooth-root),使根约束与输出可直接与 Kimodo 系列对比

模型总参数量192 M:denoiser 157 M + 动作 tokenizer 36 M(不含文本编码器),运行引擎为 PyTorch,可部署在 A100、RTX 4090 等 GPU 上。

📦 仓库结构:模型包里有什么?

该仓库是一个纯粹的模型权重包,文件结构非常清晰:

文件说明
denoiser.safetensors两阶段 Transformer 去噪器主网络权重(157M 参数)
tokenizer.safetensorsFSQ VAE 动作 tokenizer(自编码器)权重(36M 参数)
config.yaml模型结构与推理配置:骨架、帧率、FSQ 量化级别、denoiser 超参数
stats/motion/动作特征归一化统计(mean/std),含 body、global_root、local_root 子目录
stats/pre_quantization/FSQ 量化前的潜变量归一化统计
stats/post_quantization/FSQ 量化后的潜变量归一化统计

config.yaml 中几个值得关注的配置项:

  • config.yaml:gen_horizon_len: 60,正好对应模型名中的 “Horizon60”;
  • config.yaml:KimodoMotionRep运动表示 + 30 fps;num_fsq_levels: 128与fsq_level_list: 64定义 FSQ 量化码本,num_frames_per_token: 4表示每 4 帧动作被压缩成 1 个 token;
  • config.yaml:denoiser 为 8 层 8 头、latent 维度 1024 的两阶段 Transformer,trt_compatible: true表示支持 TensorRT 加速推理。

🏋️ 训练数据:630 小时动捕数据从哪来?

模型在Bones Rigplay 1数据集上完成训练与评估,核心数据信息如下(来自 README.md 模型卡):

项目详情
数据规模630 小时人类动捕数据
数据模态文本 + 人类动捕(光学动捕数据)
文本规模小于 10 亿 tokens
采集方式自动化 / 传感器采集
标注方式混合式(自动化 + 人工)
内容覆盖行走奔跑等位移、日常活动、手势等多样行为

数据预处理流水线的关键步骤:

  1. 片段切分:原始动捕数据被统一切分为10 秒片段,匹配自回归序列训练的节奏;
  2. 帧率重采样:动作统一重采样至 30 fps,与 SOMA 30 关节骨架的推理帧率对齐;
  3. 文本描述:每段动作都配有对应文本描述,覆盖位移、日常活动与手势等丰富行为。

🤖 LLM 增强:让文本标签“更多样”

数据增强的关键一步,是使用 LLM 对原始文本标签生成多样化的改写(paraphrase),扩充数据集。

这一步的价值在于:

  • 提升文本-动作配对多样性:同一段动作可对应多种自然语言描述,降低模型对单一措辞的过拟合;
  • 强化文本跟随能力:ARDY 用 R-precision 与 latent similarity 度量文本跟随准确率,训练文本越多样,语义对齐越稳健;
  • 高性价比:无需额外动捕拍摄,用 LLM 改写即可低成本扩大“有效数据量”,是动作生成模型训练中常见的技巧。

🧪 评估体系:70 小时留出集 + 三大指标

测试集设置:从 630 小时数据中留出70 小时用于测试,且测试集包含训练中未出现的内容类别——这一设计专门验证模型对未知行为的泛化能力。

根据 README.md 的 Performance Metrics,评估覆盖三个维度:

评估维度指标衡量什么
姿态约束精度(Pose Constraint Accuracy)关节距离误差生成动作对根路径、关节位置等运动学约束的遵循程度
动作质量(Motion Quality)滑步误差(foot-skating)、FID、潜变量相似度动作的物理合理性与生成分布质量
文本跟随准确率(Text-Following Accuracy)R-precision、潜变量相似度生成动作与文本提示的语义对齐程度

几个指标的含义快速上手:

  • FID:衡量生成动作与真实动作之间的分布距离,越小越好;
  • R-precision:给定文本,真实动作是否出现在生成的 Top 候选中;
  • foot-skating 误差:捕捉动作生成中最常见的物理瑕疵——脚该贴地时却“滑动”。

⚙️ 技术亮点速览

  • 架构:Novel Two-Stage Transformer 两阶段 Transformer + FSQ 量化 VAE(FSQVAETransformer),动作被压缩为离散 token 后送入去噪器;
  • 输入:文本 + 姿态约束 + 历史姿态(最多 8 秒);输出:含全局根平移与关节旋转的姿态序列(最长 8 秒);
  • 推理:TensorRT 加速,已验证 A100 与 RTX 4090 平台;
  • 生态位:ARDY 项目还有多个变体(20 fps 核心骨架、Unitree G1 机器人骨架等),本模型是其中的 SOMA 30 fps 版本。

🚀 适合谁用?典型场景与已知限制

官方用例面向任何动画经验水平的开发者与研究人员,典型场景包括:

  • 🤖 人形机器人运动规划
  • 🏭 数字孪生与工业仿真中的角色运动
  • 📊 合成数据数字人动作
  • 🎮 游戏与交互式应用动画

已知局限:生成动作可能出现滑步等伪影;模型最擅长位移、手势、格斗、舞蹈与日常活动;只能生成写实风格动作,无法产出卡通化或非物理合理的动作,且不会感知角色周围的物体。

许可条款遵循 NVIDIA Open Model Agreement,详见 README.md 模型卡;具体安装与使用方法请参考官方 ARDY 代码仓库。

  • 人工智能
  • 媒体生成
  • 数字人
  • 多模态

【免费下载链接】ARDY-SOMA-RP-30FPS-KimodoRep-Horizon60

项目地址:https://ai.gitcode.com/hf_mirrors/nvidia/ARDY-SOMA-RP-30FPS-KimodoRep-Horizon60
点击查看免费下载

相关推荐

上一篇:Wagmi 项目教程
下一篇:前端设计模式实战:Coding-Guide中的四种JavaScript设计模式详解

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

RELATED READING

延伸阅读

更多一线实战笔记与深度复盘,助您持续精进