【开源项目】ALOHA 2双手远程操作手臂-3.远程控制后收集数据

/ 858 次浏览/ 固定式Aloha

要设置新终端，请运行：

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 operation_test --episode_idx 0

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

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

vim constants.py

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']
    },
    #后面增加
        'operation_test':{
        'dataset_dir': DATA_DIR + '/operation_test',
        'num_episodes': 50,
        'episode_len': 800,
        '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/operation_test --episode_idx 0

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

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

例如：

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

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

python3 sleep.py

 aloha中文.png

    关键词：ALOHA双手远程操作手臂

    << 上一篇

    【开源项目】ALOHA 2 双手远程操作手臂-Mujoco仿真

    下一篇 >>

    【开源项目】ALOHA 2双手远程操作手臂-2.四臂联动快速上手

    相关文章