I'm developing an Autonomous Agent based on DQN. I am using the gym library to make the environments that I want to test, but I'm stuck in processing the frames of the state. The state that the gym environment returns, using the FrameStack wrapper, has the following observation space:
env = gym.make('Bowling-v0')
env = gym.wrappers.FrameStack(env, 4)
print(env.observation_space)
Box(0, 255, (4, 210, 160, 3), uint8)
I want the observation space to be Box(0, 255, (4, 88, 80, 1), uint8)
. How can I do? I've tried to use another wrapper like this:
env = gym.wrappers.ResizeObservation(env, (4, 88, 80, 1))
print(env.observation_space)
But the observation space resulting is Box(0, 255, (4, 88, 80, 1, 160, 3), uint8)
. What am I doing wrong?
Fixed! I was just doing it in the wrong way.
First must be applied the resize and then the stacking only after the resize!
So now we can proceed with the stacking with the FrameStack wrapper