issues
search
k4ntz
/
OC_Atari
Object Centric Atari games
MIT License
45
stars
8
forks
source link
general improvements
#22
Closed
sebawastaken
closed
1 year ago
sebawastaken
commented
1 year ago
OCAtari's step function now returns observations according to its obs_mode parameter
"dqn": stack of 4 grayscale images [4, 84, 84]
"ori": stack of 4 RGB images [4, 210, 160, 3]
None (default): one RGB image [210, 160, 3]
OCAtari's default obs_mode is None now (was "dqn")
object constructors in envs are consistently not expecting parameters or if they do, have default params (except riverraid)
object centric environments are available in ocatari.environments now (PositionHistoryEnv only for now) passing sb3's env_check
RL agent code for training and visualization
TODOs before merge:
[ ] check riverraid for object init (what is xfr?)
[ ] rework obs_mode values, having None as default is not nice
[ ] fix games with multiple bounding boxes for the score. Candidates (found with max_nb_objects.py):
Alien: no score object
Assault: 6 PlayerScore class instances
Asteroids: 4 PlayerScore class instances
Beamrider: not sure if hud object is score
Bowling: 2 PlayerScore class instances
Breakout: 3 PlayerScore class instances
Centipede: 4 Score class instances
ChopperCommand: 4 Score class instances
Freeway: 2 Score class instances
Frostbite: 3 PlayerScore class instances
Pitfall: 5 PlayerScore class instances
Pong: 2 EnemyScore class instances
RoadRunner: no score object
Skiing: 2 Score class instances
SpaceInvaders: 2 Score class instances
TODOs before merge: