I’m developing an RL environment which takes an image as an observation. But I don’t see any about how to use image data as an
observation_buffer in documents or examples. After I call
gymtorch.wrap_tensor, I don’t know how to convert them into the obs_buf data type like other examples correctly.
Also I see in the
train_cfg it say
policy: # only works for MlpPolicy right now. Is it mean that if i want to use an image input, I must override the script like
rl/pytorchppo/module.py, or pull the image data into a strip tensor then input as a MlpPolicy?
Can anyone give me some suggestion to solve this problem?