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

Stable Baselines Zoo には、学習済みモデルを実行した様子を動画ファイルに保存する機能があります。

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

前準備

動画ファイルを保存するために必要な ffmpeg コマンドをインストールしておきます。

sudo apt-get install ffmpeg

学習済みモデルのビデオ録画

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

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

[コマンド]

python -m utils.record_video --algo ppo2 --env MountainCar-v0 -n 1000

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