【开源项目】ALOHA 2双手远程操作手臂-【案例】方块pick up

要设置新终端,请运行:

conda activate aloha

cd ~/interbotix_ws/src/aloha/aloha_scripts

我们one_side_teleop.py跑的是测试远程操作,没有数据收集。

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

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

例如:

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

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

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

vim constants.py

Python
TASK_CONFIGS = {
    'aloha_wear_shoe':{
        'dataset_dir': DATA_DIR + '/aloha_wear_shoe',
        'num_episodes': 50,
        'episode_len': 1000,
        'camera_names': ['cam_high', 'cam_low', 'cam_left_wrist', 'cam_right_wrist']
    },
    #后面增加
        'block_pick_up':{
        'dataset_dir': DATA_DIR + '/block_pick_up',
        'num_episodes': 50,
        'episode_len': 600,
        'camera_names': ['cam_high', 'cam_low', 'cam_left_wrist', 'cam_right_wrist']
    },

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

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

 例如:

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

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

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

例如:

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

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

python3 sleep.py

 


快速收集数据

(注意:此操作是在现有数据基础上累加,比如说现在有6条数据,后面参数是50,则在6条的基础上,再收集50条数据,共56条数据)

./auto_record.sh block_pick_up 50