Aloha机械臂的mujoco仿真问题记录

本文主要是介绍Aloha机械臂的mujoco仿真问题记录，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

今天在测试ACT代码时，遇到了仿真中的机械臂无法摆放正确的姿势来抓去红色方块。

后来经过测试，发现应该是python包的版本问题有误，下面记录下正确的包版本：

官方给出的包：

conda create -n aloha python=3.8.10
conda activate aloha
pip install torchvision
pip install torch
pip install pyquaternion
pip install pyyaml
pip install rospkg
pip install pexpect
pip install mujoco==2.3.7
pip install dm_control==1.0.14
pip install opencv-python
pip install matplotlib
pip install einops
pip install packaging
pip install h5py
pip install ipython
cd act/detr && pip install -e .

直接安装官方的包，问题得到了解决！！！

Simulated experiments:

mujoco仿真数据采集指令(--num_episodes表示采集数据的数量/默认50组)：

python3 record_sim_episodes.py --task_name sim_transfer_cube_scripted --dataset_dir /media/hjx/新加卷/Linux_data/Aloha_data/act/data/sim_transfer_cube_scripted_test --num_episodes 1

可以添加标志--onscreen_render以查看实时渲染。

----------------------------------------------------------------

要在收集后可视化数据集，请运行(--episode_idx可以修改)：