sail-sg / envpool

C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.
https://envpool.readthedocs.io
Apache License 2.0
1.1k stars 100 forks source link

[Feature Request] Add Pixel-level Observations for mujoco #136

Open Benjamin-eecs opened 2 years ago

Benjamin-eecs commented 2 years ago

Motivation

https://github.com/sail-sg/envpool/pull/107#issuecomment-1123914969

Solution

Init thought:

Add method render or off_screen_render in both mujoco_env.h

Additional context

mujoco_py use MjRenderPool for multiprocess off-screen and on-screen render on multi-GPU

Road Map

Resource

dm_control

mujoco_py

FrameStack Wrapper

zhongwen commented 2 years ago

An internal project by https://github.com/Ericonaldo requires this feature. Can we bump the priority up? Thanks.