Stable Baselines Zoo - 学習済みモデルのビデオ録画(Walker2DBulletEnv-v0)

Stable Baselines Zoo の学習済みモデルを実行した様子を、動画ファイルに保存する機能を使って Walker2DBulletEnv-v0環境 を実行します。

Walker2DBulletEnv-v0環境 は2次元の下半身(Walker2D)がまっすぐ歩く(Bullet)環境です。

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

学習済みモデルのビデオ録画(Walker2DBulletEnv-v0)

次の条件で学習済みモデルのビデオ録画を実行します。

  • 環境
    Walker2DBulletEnv-v0
  • 学習アルゴリズム
    PPO2
  • ステップ数
    1000

[コマンド]

1
python3.7 -m utils.record_video --algo ppo2 --env Walker2DBulletEnv-v0 -n 1000

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


実行結果


下半身だけの物体がまっすぐ歩いている様子を確認できます。