ARDY-Core-RP-20FPS-Horizon8完全解析:革命性交互式人体运动生成模型如何重塑动画与机器人技术
ARDY-Core-RP-20FPS-Horizon8完全解析:革命性交互式人体运动生成模型如何重塑动画与机器人技术
【免费下载链接】ARDY-Core-RP-20FPS-Horizon8项目地址: https://ai.gitcode.com/hf_mirrors/nvidia/ARDY-Core-RP-20FPS-Horizon8
ARDY-Core-RP-20FPS-Horizon8是NVIDIA开发的革命性交互式人体运动生成模型,作为ARDY项目的重要组成部分,它采用自回归扩散模型架构,支持在线文本提示和灵活的长时运动学约束,为动画制作和机器人技术领域带来了突破性的解决方案。
🌟 模型简介:重新定义人体运动生成
ARDY(Autoregressive Diffusion with Hybrid Representation)是一种创新的自回归扩散模型,专为交互式运动生成设计。该模型能够响应实时文本提示,并处理多种运动学约束,包括根路径/航点、全身关键帧以及稀疏关节位置/旋转,实现了真正的实时响应能力。
ARDY-Core-RP-20FPS-Horizon8基于Bones Rigplay 1数据集训练,采用27关节"Core"骨架,运行速度为20 FPS,生成 horizon 为8帧。这一配置使其在保持高精度运动生成的同时,确保了实时交互所需的性能表现。
🚀 核心技术亮点
创新的双阶段Transformer架构
ARDY模型采用了新颖的双阶段Transformer网络架构,总参数达到326M。这一架构设计使其能够高效处理复杂的运动数据和文本提示,实现高质量的人体运动生成。
# 模型架构核心配置 [config.yaml] denoiser: _target_: ardy.model.auto_latent_twostage_denoiser.AutoLatentTwostageDenoiser latent_dim: 1024 ff_size: 2048 num_layers: 8 num_heads: 8混合表示学习
模型通过FSQ VAE Transformer实现了运动数据的高效编码与解码,结合了身体、姿态和根节点等多种特征模式,能够捕捉人体运动的细微差别和整体协调。
实时响应能力
得益于优化的网络结构和高效的推理引擎,ARDY模型能够在NVIDIA GPU上实现实时响应,这对于交互式应用场景至关重要。模型支持最大8秒的历史姿态输入和输出,满足大多数实时交互需求。
💡 强大的输入输出能力
灵活的输入方式
ARDY模型支持多种输入类型,包括:
- 文本提示(String格式)
- 姿态约束(Tensor格式)
- 历史姿态(Tensor格式)
这种多模态输入能力使得开发者可以通过自然语言描述或精确的姿态控制来生成所需的人体运动。
高质量输出
模型输出为包含全局根节点平移和关节旋转的姿态序列(Tensor格式),能够直接用于动画渲染或机器人控制。输出姿态序列最长可达8秒,以20 FPS计算相当于160帧的连续运动。
🔧 技术规格与兼容性
硬件要求
ARDY模型针对NVIDIA GPU进行了优化,支持以下微架构:
- NVIDIA Ampere
- NVIDIA Blackwell
- NVIDIA Hopper
推荐使用NVIDIA A100或RTX 4090进行推理,以获得最佳性能。
软件环境
- 运行时引擎:PyTorch
- 加速引擎:TensorRT
- 支持操作系统:Linux
模型变体
除了本项目的ARDY-Core-RP-20FPS-Horizon8外,还有其他模型变体可供选择:
- ARDY-Core-RP-20FPS-Horizon40:27关节骨架,20 FPS,40帧生成范围
- ARDY-G1-RP-25FPS-Horizon52:34关节Unitree G1机器人骨架,25 FPS,52帧生成范围
- ARDY-G1-RP-25FPS-Horizon8:34关节Unitree G1机器人骨架,25 FPS,8帧生成范围
📊 训练数据与性能
训练数据集
模型基于Bones Rigplay 1数据集进行训练,包含:
- 630小时的人体运动捕捉数据
- 对应的文本描述
- 涵盖多种行为: locomotion、日常活动和手势
- 运动片段时长为10秒,并按所需FPS重采样
评估指标
ARDY模型的性能通过以下指标进行评估:
- 姿态约束精度(关节距离误差)
- 运动质量(足部滑动误差、FID、潜在空间相似度)
- 文本跟随精度(R-precision、潜在空间相似度)
💻 快速开始:使用指南
要开始使用ARDY-Core-RP-20FPS-Horizon8模型,请按照以下步骤操作:
- 克隆仓库:
git clone https://gitcode.com/hf_mirrors/nvidia/ARDY-Core-RP-20FPS-Horizon8参考官方代码库获取详细的安装和使用说明:https://github.com/nv-tlabs/ardy
模型文件位于项目根目录:
- 去噪器模型:denoiser.safetensors
- 分词器模型:tokenizer.safetensors
- 配置文件:config.yaml
- 统计数据:stats/
🚀 应用场景
ARDY模型为多种应用场景提供了强大的人体运动生成能力:
动画制作
无论是游戏开发还是影视制作,ARDY都能快速生成自然流畅的角色动画,大大减少动画师的工作量。开发者可以通过简单的文本描述或关键帧控制,生成复杂的人物动作。
机器人技术
对于人形机器人,ARDY模型可用于运动规划,帮助机器人实现更自然、更灵活的移动方式。特别是Unitree G1机器人专用模型变体,为机器人控制提供了量身定制的解决方案。
数字孪生与工业模拟
在数字孪生系统和工业模拟中,ARDY可以生成逼真的人体运动,用于培训模拟、流程优化等场景,提高模拟的真实性和实用性。
合成数据生成
ARDY模型能够生成大量多样化的人体运动数据,可用于训练其他计算机视觉模型,如人体姿态估计、动作识别等系统。
⚠️ 注意事项与限制
在使用ARDY模型时,请注意以下限制:
生成的运动可能包含一些不自然的 artifacts,如足部滑动(当脚应该与地面保持静态接触时却发生滑动)。
模型并不总是能完全遵循给定的文本提示,对于某些类型的动作可能表现不佳(模型最擅长locomotion、手势、战斗、舞蹈和日常活动)。
每个训练好的模型目前只能输出单一角色骨架的运动。
模型设计用于输出逼真的人体运动,因此无法创建卡通风格或不符合物理规律的运动。
模型不了解场景中角色周围的物体。
📝 许可证信息
ARDY-Core-RP-20FPS-Horizon8模型的使用受NVIDIA Open Model Agreement管辖。该模型可用于商业或非商业用途,具体条款请参见NVIDIA Open Model Agreement。
📚 参考资料
- 论文:ARDY: Autoregressive Diffusion with Hybrid Representation for Interactive Human Motion Generation
- 项目页面:https://research.nvidia.com/labs/sil/projects/ardy/
- 代码库:https://github.com/nv-tlabs/ardy
- 训练数据集:Bones Rigplay 1 dataset
ARDY-Core-RP-20FPS-Horizon8模型代表了交互式人体运动生成领域的重大进步,它不仅为开发者和研究人员提供了强大的工具,也为动画制作和机器人技术等行业开辟了新的可能性。无论您是经验丰富的开发者还是刚入门的新手,都能通过这个模型轻松创建出自然、流畅的人体运动效果。
【免费下载链接】ARDY-Core-RP-20FPS-Horizon8项目地址: https://ai.gitcode.com/hf_mirrors/nvidia/ARDY-Core-RP-20FPS-Horizon8
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
