mujoco graspnet 仿真项目的复现记录

开源项目:https://gitee.com/chaomingsanhua/manipulator_grasp

复现使用的配置:linux系统ubuntu20.04


项目配置记录:

git clone 对应的code后:

需要在graspnet-baseline文件夹中继续拉取文件,指令记录:

cd manipulator_grasp

git clone https://github.com/graspnet/graspnet-baseline.git

下面需要继续配置graspnet-baseline项目的环境:

这里为了这个项目单独创建了一个conda环境

conda create -n graspnet python=3.9

conda activate graspnet

pip install graspnetAPI

pip uninstall trimesh

pip install trimesh==3.9.0

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

pip install spatialmath-python

pip install mujoco

pip install modern-robotics

pip install roboticstoolbox-python

pip install numpy==1.23.0

Compile and install pointnet2 operators (code adapted from votenet).

cd pointnet2

python setup.py install

Compile and install knn operator (code adapted from pytorch_knn_cuda).

cd knn

python setup.py install


下面需要下载该项目对应的 checkpoint-rs.tar 文件(在graspnet-baseline项目的Readme中):

将其存放在该路径下:

下面开始运行main.py主程序代码:

会遇到如下报错:
Traceback (most recent call last): File "/home/song/hjx/manipulator_grasp/main.py", line 19, in <module> from graspnet_dataset import GraspNetDataset File "/home/song/hjx/manipulator_grasp/graspnet-baseline/dataset/graspnet_dataset.py", line 12, in <module> from torch._six import container_abcs # pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html ModuleNotFoundError: No module named 'torch._six'

修复步骤​

python 复制代码
# 原代码(旧版PyTorch)
from torch._six import container_abcs

# 修改为(适配PyTorch 2.x)
import collections.abc as container_abcs

继续运行main.py主程序代码,会出现如下界面:

关闭这个图片的画面,机械臂开始抓取运动:

mujoco仿真的效果十分nice!!!


参考:
Ubuntu20.04复现GraspNet全记录(含遇到的问题及解决方法_graspnet数据集下载-CSDN博客

https://github.com/graspnet/graspnet-baseline

相关推荐
reasonsummer10 小时前
【办公类-04-05】20261030整理手机照片(华为手机助手导出照片到日期文件夹+批量添加四级文件夹+手动转移照片更名+更改三级文件夹合并汉字名称)
python·智能手机
服务器硬件项目经理10 小时前
《超节点修仙录》· 第一卷·炼气期【第04话】 超节点 vs 普通服务器——不就是多插了几张卡吗?
运维·服务器·人工智能·嵌入式硬件·深度学习·机器学习
BD_Marathon11 小时前
【Python】初识迭代器
python
gz927cool11 小时前
Python 全栈应用最佳实践完整指南
开发语言·python·ai
一切皆是因缘际会11 小时前
当模型足够大
人工智能·深度学习·机器学习
智购科技智能售货柜11 小时前
顾客已经关门,视觉售货柜还在处理旧画面:延迟积压怎么查?~YH
python·单片机·嵌入式硬件·物联网·eclipse
梯度下降者11 小时前
Python从0到100(三):Python中的变量介绍
python·作用域·变量·数据类型·计算机常识
AIFQuant11 小时前
Python期货行情API接入:股指、黄金、原油多品种数据管道实战
开发语言·python·金融·区块链·量化开发
威哥爱编程12 小时前
Python 智能体开发实战 07:让它查课表它张口就编?给 Agent 接上真正能动手的工具
python·agent
Wang's Blog12 小时前
Java 项目实战:交友项目 - 即时通信总结:全链路复盘、可复用清单与自检
java·python·交友