解决:ubuntu22.04中IsaacGymEnv保存视频报错的问题

news/2025/1/17 13:39:20/

1. IsaacGymEnvs项目介绍

IsaacGymEnvs:基于NVIDIA Isaac Gym的高效机器人训练环境

IsaacGymEnvs 是一个基于 NVIDIA Isaac Gym 的开源 Python 环境库,专为机器人训练提供高效的仿真环境。Isaac Gym 是由 NVIDIA 开发的一个高性能物理仿真引擎,专为机器人学和强化学习等应用设计。它利用 GPU 加速,支持大规模的并行仿真,使得多智能体强化学习和机器学习任务的训练速度大大提升。

1.1. 主要特点

  1. GPU 加速 : Isaac Gym 使用 GPU 进行并行计算,可以高效地模拟多个环境,支持快速的仿真训练,特别适合强化学习和深度学习任务。
  2. 多智能体支持IsaacGymEnvs 提供了多种多智能体环境的支持,适合训练多个机器人或智能体进行协作或竞争。
  3. 高性能 : 相比于传统的基于 CPU 的仿真引擎,Isaac Gym 通过 GPU 的并行计算,极大地提高了仿真速度和训练效率。
  4. 适用于强化学习IsaacGymEnvs 设计支持与强化学习框架(如 RLlib、Stable Baselines3、PyTorch 和 TensorFlow)集成,方便用户进行机器人控制、策略优化等任务。
  5. 物理引擎和仿真 : 它结合了 NVIDIA 物理引擎,提供了高精度的碰撞检测、物体交互、重力模拟等功能,适用于训练基于物理仿真的任务。
  6. 丰富的环境 : 提供了多种预定义的环境,如操控任务、导航任务、物体抓取等,可以作为快速原型设计和实验的基础。
  7. 易于扩展 : 用户可以创建自定义的环境,通过修改现有的环境或开发新环境来满足特定的需求。

1.2. 常见使用场景

  • 机器人控制:训练机器人在模拟环境中执行复杂的任务,如抓取、操作物体、导航等。
  • 强化学习:开发和训练基于强化学习的算法,进行自我学习和优化策略。
  • 多智能体协作与对抗:训练多个机器人或智能体进行协作、对抗或竞争任务。
  • 物理仿真研究:研究机器人与环境交互中的物理现象,优化机器人设计和控制策略。

1.3. 集成与使用

IsaacGymEnvs 可以与多个流行的强化学习库结合使用,如:

  • RLlib:一个强化学习库,能够处理分布式强化学习任务。
  • Stable Baselines3:一个用于强化学习的高层次库,提供简单的接口和算法实现。
  • PyTorchTensorFlow:深度学习框架,可以与 IsaacGymEnvs 进行无缝集成,用于训练深度强化学习模型。

2. 保存视频报错及修改

isaac gym实现了标准 env.render(mode='rgb_rray') gym API 来提供模拟器查看器的图像。 此外,可以利用gym.wrappers.RecordVideo来帮助录制显示智能体运动过程的视频。 官方给出了示例代码如下,该文件应在视频文件夹中生成视频。

假如已经训练好了,用预训练模型参数保存视频的代码是:

python train.py task=FrankaCubeStack test=True checkpoint=runs/FrankaCubeStack_09-12-14-22/nn/last_FrankaCubeStack_ep_500_rew_773.4196.pth headless=False num_envs=64 capture_video=True

2.1. 提示缺包:

在这里插入图片描述
重装pillow:

 sudo apt-get install xvfb 

2.2. 继续报错:

在这里插入图片描述
OSError: Pillow was built without XCB support 表示当前环境中的 Pillow(Python 的图像处理库)不支持 XCB 功能,可能与图像显示或处理环境不兼容。

卸载、重装pillow:

pip uninstall pillow
pip install pillow

在这里插入图片描述

2.3. 继续报错:

Unknown encoder 'libx264'
ERROR: VideoRecorder encoder failed: None
ERROR: VideoRecorder encoder failed: None

在这里插入图片描述此时可以生成视频,但是视频无法写入图像内容,视频大小是0 bytes:在这里插入图片描述

查看系统是否安装了 ffmpeg,ffmpeg是否支持 libx264 编码器:

ffmpeg -encoders | grep libx264

在这里插入图片描述
问题分析: 从输出信息来看,ffmpeg 已经安装,但是 libx264 编码器没有被启用,说明当前的 ffmpeg 配置没有包含 libx264 编码支持。
解决方法: 需重新编译 ffmpeg,并确保启用 libx264 编码器。

conda install -c conda-forge ffmpeg

在这里插入图片描述

2.4. 成功写出视频

再次运行

python train.py task=FrankaCubeStack test=True checkpoint=runs/FrankaCubeStack_09-12-14-22/nn/last_FrankaCubeStack_ep_500_rew_773.4196.pth headless=False num_envs=64 capture_video=True

成功写出视频:
在这里插入图片描述

rl-video-step-0


http://www.ppmy.cn/news/1563891.html

相关文章

【VitePress快速入门】页面导航nav与侧边栏siderbar配置(三)

前期回顾 【VitePress快速入门】初始化项目(一)-CSDN博客 【VitePress快速入门】项目配置(网站图标、更新时间、中文化等)(二)-CSDN博客 1.导航栏与侧边栏 导航栏与侧边栏分别由config.mjs中的nav和sider…

如何设置请求头模拟浏览器访问?

在Python中使用requests库发送HTTP请求时,可以通过设置请求头(Headers)来模拟浏览器访问。这有助于避免被目标网站识别为爬虫而被阻止访问。以下是如何在代码示例中设置请求头的详细步骤和解释。 一、设置请求头 请求头(Headers…

C++实现设计模式---中介者模式 (Mediator)

中介者模式 (Mediator) 中介者模式 是一种行为型设计模式,它用一个中介对象来封装一组对象之间的交互。中介者通过协调多个对象之间的通信,避免对象之间的直接依赖,从而实现对象之间的松耦合。 意图 通过引入一个中介者对象,减少…

如何学习Transformer架构

Transformer架构自提出以来,在自然语言处理领域引发了革命性的变化。作为一种基于注意力机制的模型,Transformer解决了传统序列模型在并行化和长距离依赖方面的局限性。本文将探讨Transformer论文《Attention is All You Need》与Hugging Face Transform…

细说STM32F407单片机窗口看门狗WWDG的原理及使用方法

目录 一、窗口看门狗的工作原理 1、递减计数器 2、窗口值和比较器 3、看门狗的启动 4、提前唤醒中断 二、窗口看门狗的HAL驱动程序 1、窗口看门狗初始化 2.窗口看门狗刷新 3.EWI中断及其处理 三、不开启EWI的WWDG示例 1、示例功能 2、项目设置 (1&…

Vue 和 uniApp 中 CSS 样式差别

之前一直在做vue2的项目,最近开始uniapp的项目,发现两种项目之间css还是有亿点区别的。 一、布局单位 Vue 2 项目: 通常使用 px 作为主要的长度单位,这是一个绝对单位,在不同设备屏幕上显示的物理尺寸相同。例如&am…

备忘录清单工具:四款电脑桌面备忘录工具分享

在快节奏的现代生活中,无论是工作还是学习,我们每天都会面临各种各样的任务和待办事项。为了更好地管理这些繁杂的事务,一款好用的电脑桌面备忘录工具无疑是提升效率的关键。今天,就让我们一起来看看四款备受好评的备忘录工具。 …

30分钟内搭建一个全能轻量级springboot 3.4 + 脚手架 <5> 5分钟集成好caffeine并使用注解操作缓存

快速导航 <1> 5分钟快速创建一个springboot web项目 <2> 5分钟集成好最新版本的开源swagger ui&#xff0c;并使用ui操作调用接口 <3> 5分钟集成好druid并使用druid自带监控工具监控sql请求 <4> 5分钟集成好mybatisplus并使用mybatisplus generator自…