AnyGrasp真机部署

  • 登录github项目:https://github.com/graspnet/anygrasp_sdk?tab=readme-ov-file
  • 创建conda环境
  • conda create --name anygrasp python=3.9
    conda activate anygrasp
  • 下面准备安装依赖项目MinkowskiEngine:https://github.com/NVIDIA/MinkowskiEngine#anaconda
  • 检查cuda版本,要求11.x。本人版本11.7。
  • nvcc -V
  • 进入pytorch官网下载torch(推荐开梯子官网下载):Previous PyTorch Versions | PyTorch
  • 本人选择CUDA11.7,因此找到对应安装命令,conda环境下进行安装
  • conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia
  • 检查是否安装成功
  • python
    >>> import torch
    >>> torch.cuda.is_available()
    True
    >>> quit()
    
  • 开始正式安装MinkowskiEngine
  • git clone https://github.com/NVIDIA/MinkowskiEngine.git
    cd MinkowskiEngine
    conda clean -all  #清除conda缓存,防止重复下载torch
    pip install ninja  #官方文档没说,这里依赖还需要安装ninja库
    conda install openblas-devel -c anaconda  #安装依赖
    export CUDA_HOME=/usr/local/cuda-11.x  #你的版本
    export MAX_JOBS=2
    cp  /home/xxx(username)/anaconda3/envs/anygrasp/lib/libopenblas.so*  /home/xxx(username)/anaconda3/envs/anygrasp/lib/python3.9/site-packages/torch/lib/.
    python setup.py install --blas_include_dirs=${CONDA_PREFIX}/include --blas=openblas
  • MinkowskiEngine安装成功,检查。
  • python
    >>> import MinkowskiEngine as ME
    >>> print(ME.__version__)
    0.5.4
  • 回到anygrasp项目,开始正式安装anygrasp
  • 安装依赖
  • export SKLEARN_ALLOW_DEPRECATED_SKLEARN_PACKAGE_INSTALL=True
    pip install -r requirements.txt
    cd pointnet2
    python setup.py install
    cd ..
  • 申请SDK的license和checkpoint,根据仓库readme提交表格申请。会收到三份文件。分别是,license文件(需要解压),两个checkpoint文件(用于进行任务)。
  • 解压license后将文件夹复制两份,重命名为license,分别放在 grasp_detection 和 grasp_tracking 下。
  • 演示grasp_detection。
  • 进入grasp_detection文件夹。
  • 根据您的 Python 版本(支持 Python>=3.6,<=3.9),将 gsnet.*.solib_cxx.*.so 复制到此文件夹。例如,如果您使用 Python 3.6,则可以执行以下操作:
  • cp gsnet_versions/gsnet.cpython-36m-x86_64-linux-gnu.so gsnet.so
    cp ../license_registration/lib_cxx_versions/lib_cxx.cpython-36m-x86_64-linux-gnu.so lib_cxx.so
  • 其他python版本见目录。
  • 在grasp_detection文件夹下创建log文件夹,将模型权重checkpoint_detection.tar放在log/下。
  • 运行:
  • sh demo.sh
  • 成功输出

  • 可视化运行:

  • python demo.py  --checkpoint_path ./log/checkpoint_detection.tar --debug
  • 爪子备选,可以拖动放大缩小进行观察。

  • 按下q,获得最终爪子标记。

  • 结束。grasp_tracking同理。运行后动态展示跟踪。

  • 如果你觉得有帮助的话,请给一个小小的赞!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

花工靴子

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值