Stable Baselines Zoo には、学習済みモデルを実行している様子を動画ファイルに保存する機能があります。
この機能を使って AntBulletEnv-v0環境 を実行する様子を動画ファイルに出力します。
AntBulletEnv-v0環境 はアリっぽい形をした４本足の模型がうまく歩けるように調整する環境です。
(Ubuntu 19.10で動作確認しています。)
学習済みモデルの動画ファイル出力(AntBulletEnv-v0)
次の条件で、学習済みモデルを実行している様子を動画ファイルに出力します。
- 環境
AntBulletEnv-v0 - 学習アルゴリズム
PPO2 - ステップ数
1000
[コマンド]
1 | python3.7 -m utils.record_video --algo ppo2 --env AntBulletEnv-v0 -n 1000 |
上記コマンドを実行すると、次のような動画ファイルが出力されます。
きちんと（？）虫っぽい動きで歩く様子を確認することができます。