OpenAI Gymのインベーダーゲームをやってみる

使ったリポジトリ

実行

git clone https://github.com/tensorpack/tensorpack.git
cd example/A3C-Gym

train-atari.pyに下記のコードを追加しパスを通す

import sys
sys.path.append("../../")

学習済みモデルをインストール

!wget http://models.tensorpack.com/OpenAIGym/SpaceInvaders-v0.tfmodel

テストする

!python ./train-atari.py --task dump_video --load SpaceInvaders-v0.tfmodel --env SpaceInvaders-v0 --output SpaceInvaders-v0 --episode 3

--task dump_video 動画を保存する
--load ~~ 使用する学習済みモデル
--env ~~ 使用するゲーム　今回の場合はスペースインベーダー
--output ~~ 動画を保存するディレクトリ(実行前にそのディレクトリが存在するとエラーとなる)

結果