Kanazawanaoaki / jsk_learning_utils

4 stars 1 forks source link

Refactor personal scripts #4

Closed HiroIshida closed 2 years ago

HiroIshida commented 2 years ago

実行スクリプト群のリファクタリング.

コメント

実装とは関係ないコメント

train_LSTMとtrain_CAE.py で すべてのrosbagを使って訓練してると思うのですが, test_CAE でもtest_LSTMでも 訓練時のロスバグと同じものを使ってテストしている(画像とか動画とかを作ってるので)のでこのままだとテストになってないかも... あまり詳細をみてないので間違ってたらすみません。

実装のお願い(練習もかねて)

金沢独自スクリプト群を本PRに対応させてみて、このブランチに対してPRを投げてみてください

少しややこしいし難しい(変更点も多い)のでこちらは僕の方でやっておきました。

HiroIshida commented 2 years ago

とりあえず金沢くんのスクリプトを https://github.com/Kanazawanaoaki/jsk_learning_utils/pull/4/commits/2616ff8bed04dd65c1f444f6a17e2ed7f3e96df4 で本PRに対応させました。

rosrun jsk_learning_utils test_kanazawa_learning.sh を実行すると学習からテストまで終わり, project_data/sample_rcup_pick/movies の以下に結果の動画がダンプされます. 以下が結果なのでたぶんうまくいっているようです. もしかしたらgdownのスクリプトでaccess deniedがでるかもしれませんが, これはgoogle drive 側のバグのようです。https://github.com/wkentaro/gdown/pull/42 エラーが出た場合は、手動でzipファイルをダウンロードしてください

lstm gif rosbag-1645162613114649057

reconstrcution gif rosbag-1645162613114649057_reconstruction

Kanazawanaoaki commented 2 years ago

他の例にも対応可能にするために,

をやってみます.

configファイルの仕様

基本的にはjsk_learning_utils/configs/以下のconfig.yamlを使う.
それ以外のconfigを使う時にはhogehoge.yamlを追加して引数で指定する.
どちらの場合もconfigファイルをproject_dataのプロジェクトファイルにコピーして確認できる&実行時に読み込めるようにする.