Farama-Foundation / Arcade-Learning-Environment

The Arcade Learning Environment (ALE) -- a platform for AI research.
GNU General Public License v2.0
2.12k stars 420 forks source link

Is it standard to grayscale? #458

Closed slerman12 closed 2 years ago

slerman12 commented 2 years ago

I've been using rgb.

gym.make(env, obs_type='rgb')

However, StableBaselines3 uses grayscale.

Which one is preferred/standard?

Thanks.

JesseFarebro commented 2 years ago

Hi @slerman12, the standard methodology since the release of DQN is that the agent will observe grayscaled frames, down sampled to 84x84 and stacked in the depth channel with the previous 4 observations.

There's a lot of nuance in the preprocessing stack for the ALE. I would highly suggest using something that's tried and tested. I would recommend looking at:

We provide the option to retrieve RGB observations from gym.make but in the future I plan on building the complete preprocessing pipeline into the ALE itself so there isn't all these different postprocessing implementations.

Let me know if you have any other questions about this, I know it can be confusing, the most common questions I receive are around postprocessing.