diff --git a/BattlePongAI/main.py b/BattlePongAI/main.py index 25bb3fd9e163d899d8b1cfe9d510928dee0e9d25..9b5f2856c63f74457273727287fb81fe1edc1c99 100644 --- a/BattlePongAI/main.py +++ b/BattlePongAI/main.py @@ -15,7 +15,8 @@ async def main(): recieved_values = await gym.reset() data = recieved_values[0] print('main data', data) - image_mode = 'RGB' + #image_mode = 'RGB' + image_mode = 'L' try: # Test show Image si.show_image(image_mode, data['info']['screenshot']['width'], data['info']['screenshot']['height'],