首页
/
行业洞察
/
正文
INDUSTRY INSIGHT · 深度
MuJoCo中运行和导出ONNX模型
📅 2026/9/6 7:21:56
✍️ 爱科研究院
👁 阅读 3,247
在MuJoCo中运行和导出ONNX模型通常是强化学习工作流中的两个关键步骤导出是将训练好的策略模型转换为通用格式运行则是在仿真环境中验证这个模型。整个过程可以总结为“训练 - 导出 - 仿真验证”下面以你之前接触的microduck_rl项目为例进行说明。 导出ONNX模型将策略变为通用文件导出过程的核心是将训练好的神经网络策略通常是.pt文件转换为.onnx格式并烘焙Bake观测归一化器。这一步至关重要它确保了模型在实际部署时接收的数据和处理方式与训练时完全一致。核心命令在microduck_rl项目根目录下使用官方导出脚本。bashuv run scripts/export.py Mjlab-Velocity-Flat-MicroDuck --wandb-run-path 你的wandb运行路径重要说明必须使用官方脚本microduck_rl的export.py脚本不仅导出网络权重还会将观测归一化器的参数固化到ONNX模型中。手动转换会导致策略在仿真或真机上失效。输入需要指定训练好的模型路径通常通过--wandb-run-path指向记录训练过程的WandB项目。输出命令执行后会在指定目录生成一个.onnx文件例如output.onnx这就是后续要运行的文件。▶️ 运行ONNX模型在仿真中验证策略运行ONNX模型就是在MuJoCo仿真环境中加载并执行这个策略观察机器人的表现这被称为“仿真彩排”。核心命令同样在microduck_rl项目根目录下使用推理脚本。bashuv run scripts/infer_policy.py --walking output.onnx重要说明CPU即可运行这一步对计算资源要求不高在普通电脑的CPU上就能运行。交互与控制运行后会弹出一个MuJoCo仿真窗口。你可以通过键盘来控制MicroDuck例如让它行走、踢球等。不同的命令对应不同的动作--help参数可以查看所有选项。功能扩展该脚本还支持--record录像和--save-csv导出数据方便进行仿真到现实Sim2Real的对比分析。 总结总的来说这个流程非常清晰导出在microduck_rl项目中使用uv run scripts/export.py ...命令将训练好的策略如.pt文件导出为包含归一化参数的.onnx文件。运行在同一项目中使用uv run scripts/infer_policy.py --walking 你的模型.onnx命令在MuJoCo仿真中加载并测试导出的策略。这个“导出-运行”的循环是验证策略有效性的关键一步也是从仿真走向真实机器人的必经之路。
📌 标签:
工业官网
设计趋势
AI 建站
SEO
获取完整报告 →
RELATED ARTICLES
推荐阅读
2026/9/6 7:21:56
被AIGC检测打回两次后,我终于理清论文AI工具分工:大模型想思路、科研工具找文献、毕业之家降痕迹
2026/9/6 7:21:56
最新GEO平台排名前十怎么选才不踩坑:从技术底座到落地效果的全景对比
2026/9/6 7:16:56
应收账款分析有哪些坑?应收账款分析怎么避免出错?
2026/9/6 8:46:59
STM32F407VET6实战详解:引脚、最小系统、PHY与例程
2026/9/6 8:46:59
FreeRTOS实战指南:任务调度、内存管理与堆栈溢出排查
2026/9/6 8:46:59
企业应如何根据自身需求选择人力资源服务?
2026/9/6 8:46:59
API 2.0开发实战:错误处理与CPI性能监控最佳实践
2026/9/6 8:46:59
STM32智能温控风扇从入门到实战:方案、电路与代码全解析
2026/9/6 8:41:59
Apifox管理大模型接口:环境切换与自动化测试实战指南
2026/9/6 0:01:31
超人会飞不算本事:系统稳定依赖清晰规则与边界设计
2026/9/6 0:01:31
超人VS蜘蛛侠:拆解超级IP的影响力与传播方法论
2026/9/6 0:01:31
基于CNN的调制信号识别:MATLAB实现时频图分类实战
2026/9/6 0:01:31
超人会飞不算本事:系统稳定依赖清晰规则与边界设计
2026/9/6 0:01:31
超人VS蜘蛛侠:拆解超级IP的影响力与传播方法论
2026/9/6 0:01:31
基于CNN的调制信号识别:MATLAB实现时频图分类实战