Stable Baselines Zoo - 学習済みモデルを動画ファイルに出力(sPacmanNoFrameskip-v4)

Stable Baselines Zoo の動画ファイル出力機能を使って sPacmanNoFrameskip-v4環境 を実行する様子を動画ファイルにしてみます。

sPacmanNoFrameskip-v4環境 は、あの有名なパックマンです。

(Ubuntu 19.10で動作確認しています。)

学習済みモデルの動画ファイル出力(sPacmanNoFrameskip-v4)

次の条件で、学習済みモデルを実行している様子を動画ファイルに出力します。

  • 環境
    sPacmanNoFrameskip-v4
  • 学習アルゴリズム
    DQN
  • ステップ数
    1000

[コマンド]

1
python3.7 -m utils.record_video --algo dqn --env sPacmanNoFrameskip-v4 -n 1000

上記コマンドを実行すると、次のような動画ファイルが出力されます。


実行結果


下手な人がパックマンをやっているような感じがします。

強化学習を通してパックマンがさくさくクリアするくらいになるとかなり嬉しいというか実用的なのですが、私の知識ではまだまだそこまでいきません。m(_ _ )m