Moblie-Aloha家务机器人-D.Mobile-aloha数据集收集

要收集某一集的数据,请运行:

python3 record_episodes.py --task_name <your task_name> --episode_idx 0

例如:

python3 record_episodes.py --task_name aloha_mobile_dummy --episode_idx 0

这将在<data save dir>处存储hdf5文件

要更改剧集长度和其他参数,请直接编辑constants.py。

vim constants.py

要可视化收集的剧集,请运行:

python3 visualize_episodes.py --dataset_dir <data save dir> --episode_idx 0

 例如:

python3 visualize_episodes.py --dataset_dir ~/data/aloha_mobile_dummy --episode_idx 0

要重播使用真实机器人收集的剧集,请运行:

python3 replay_episodes.py --dataset_dir <data save dir> --episode_idx 0

例如:

python3 replay_episodes.py --dataset_dir ~/data/aloha_mobile_dummy --episode_idx 0

要在切断电源之前降低4个机器人,请运行:

python3 sleep.py