Naereen / gym-nes-mario-bros

🐍 🏋 OpenAI GYM for Nintendo NES emulator FCEUX and 1983 game Mario Bros. + Double Q Learning for mastering the game
https://naereen.github.io/gym-nes-mario-bros/
MIT License
49 stars 15 forks source link

Use difference of screen instead of screen as input to the DQN #1

Closed Naereen closed 6 years ago

Naereen commented 6 years ago

https://github.com/Naereen/gym-nes-mario-bros/blob/518d89b6635a438bbe3c00d0f84fe06aafd61ed6/src/nesgym/nesenv.py#L243 Should be

obs = self.screen.copy() - self.last_screen
self.last_screen = self.screen.copy()