Atari

This dataset comprises expert demonstrations across 57 distinct Atari games, showcasing a wide range of gameplay dynamics and difficulty levels. For each environment, we recorded 10 complete episodes using pretrained agents from CleanRL. Although the agents operated with standard preprocessing techniques, such as frame stacking, grayscale conversion, and image resizing, the dataset preserves raw images as observations, allowing users the flexibility to apply their own preprocessing methods.

Content

ID	Description
Alien	You are stuck in a maze-like space ship with three aliens.
Amidar	This game is similar to Pac-Man: You are trying to visit all places on a 2-dimensional grid while simultaneously avoiding your enemies.
Assault	You control a vehicle that can move sideways.
Asterix	You are Asterix and can move horizontally (continuously) and vertically (discretely).
Asteroids	This is a well-known arcade game: You control a spaceship in an asteroid field and must break up asteroids by shooting them.
Atlantis	Your job is to defend the submerged city of Atlantis.
Bank Heist	You are a bank robber and (naturally) want to rob as many banks as possible.
Battle Zone	You control a tank and must destroy enemy vehicles.
Beam Rider	You control a space-ship that travels forward at a constant speed.
Berzerk	You are stuck in a maze with evil robots.
Bowling	Your goal is to score as many points as possible in the game of Bowling.
Boxing	You fight an opponent in a boxing ring.
Breakout	Another famous Atari game.
Centipede	You are an elf and must use your magic wands to fend off spiders, fleas and centipedes.
Chopper Command	You control a helicopter and must protect truck convoys.
Crazy Climber	You are a climber trying to reach the top of four buildings, while avoiding obstacles like closing windows and falling objects.
Defender	Aliens attack the earth.
Demon Attack	You are facing waves of demons in the ice planet of Krybor.
Double Dunk	You are playing a 2v2 game of basketball.
Enduro	You are a racer in the National Enduro, a long-distance endurance race.
Fishing Derby	Your objective is to catch more sunfish than your opponent.
Freeway	Your objective is to guide your chicken across lane after lane of busy rush hour traffic.
Frostbite	In Frostbite, the player controls “Frostbite Bailey” who hops back and forth across across an Arctic river, changing the color of the ice blocks from white to blue.
Gopher	The player controls a shovel-wielding farmer who protects a crop of three carrots from a gopher.
Gravitar	The player controls a small blue spacecraft.
Hero	You need to rescue miners that are stuck in a mine shaft.
Ice Hockey	Your goal is to score as many points as possible in a standard game of Ice Hockey over a 3-minute time period.
Jamesbond	Your mission is to control Mr.
Kangaroo	The object of the game is to score as many points as you can while controlling Mother Kangaroo to rescue her precious baby.
Krull	Your mission is to find and enter the Beast’s Black Fortress, rescue Princess Lyssa, and destroy the Beast.
Kung Fu Master	You are a Kung-Fu Master fighting your way through the Evil Wizard’s temple.
Montezuma Revenge	Your goal is to acquire Montezuma’s treasure by making your way through a maze of chambers within the emperor’s fortress.
Ms PacMan	Your goal is to collect all of the pellets on the screen while avoiding the ghosts.
Name This Game	Your goal is to defend the treasure that you have discovered.
Phoenix	Your goal is to reach and shoot the alien pilot.
Pitfall	You control Pitfall Harry and are tasked with collecting all the treasures in a jungle within 20 minutes.
Pong	You control the right paddle, you compete against the left paddle controlled by the computer.
Private Eye	You control the French Private Eye Pierre Touche.
Qbert	You are Q*bert.
Riverraid	You control a jet that flies over a river: you can move it sideways and fire missiles to destroy enemy objects.
Road Runner	You control the Road Runner(TM) in a race; you can control the direction to run in and times to jumps.
Robotank	You control your Robot Tanks to destroy enemies and avoid enemy fire.
Seaquest	You control a sub able to move in all directions and fire torpedoes.
Skiing	You control a skier who can move sideways.
Solaris	You control a spaceship.
Space Invaders	Your objective is to destroy the space invaders by shooting your laser cannon at them before they reach the Earth.
Star Gunner	Stop the alien invasion by shooting down alien saucers and creatures while avoiding bombs.
Surround	Surround your opponent without running into anything yourself.
Tennis	You control the orange player playing against a computer-controlled blue player.
Time Pilot	You control an aircraft.
Tutankham	Your goal is to rack up points by finding treasures in the mazes of the tomb while eliminating its guardians.
Up n Down	Your goal is to steer your baja bugger to collect prizes and eliminate opponents.
Venture	Your goal is to capture the treasure in every chamber of the dungeon while eliminating the monsters.
Video Pinball	Your goal is to keep the ball in play as long as possible and to score as many points as possible.
Wizard of Wor	Your goal is to beat the Wizard using your laser and radar scanner.
Yars Revenge	The objective is to break a path through the shield and destroy the Qotile with a blast from the Zorlon Cannon.
Zaxxon	Your goal is to stop the evil robot Zaxxon and its armies from enslaving the galaxy by piloting your fighter and shooting enemies.