Graspnet視覺抓取(一)——環境搭建
前言
·本文是關于Graspnet的環境搭建教程。·不同于一般因為Graspnet庫過老,只能建立python<3.10的情況,本文使用python=3.10、CUDA=12.1進行環境搭建。
·Graspnet的開源代碼鏈接:https://github.com/graspnet/graspnet-baseline
一、環境搭建
### 1.下載Graspnet開源代碼git clone https://github.com/graspnet/graspnet-baseline.git
這步如果有VPN,但還存在問題,要么打開瀏覽器去Graspnet的github上下載ZIP,要么去看我這篇博客:Linux添加終端代理:http://www.rzrgm.cn/john-mu-wanfeng/p/19078263
下載后在終端當前路徑下可以找到該文件夾:

2.創建虛擬環境
這步需要使用到Anaconda,如果你使用其他的虛擬環境軟件,仿照我的步驟即可,否則搜索Anaconda的安裝教程,進行安裝后再執行后續步驟。
conda create -n graspnet python=3.10
conda activate graspnet
cd graspnet-baseline
3.安裝Pytorch三件套
由于我的CUDA版本是12.1,因此這里我就按照12.1的版本繼續安裝了,運行如下命令即可:
conda install pytorch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 pytorch-cuda=12.1 -c pytorch -c nvidia
如果你的電腦不是CUDA-12.1,你可以按照你的CUDA版本來,或者參考我這篇博客Linux多版本CUDA切換:http://www.rzrgm.cn/john-mu-wanfeng/p/19071862下載CUDA-12.1,這樣后續步驟跟我一模一樣即可。
4.安裝Graspnet依賴庫
這里步驟基本上與官網類似,只是有些文件里寫的庫的版本或者名稱需要修改。
(修改1).修改graspnet-baseline文件夾中的requirements.txt:

將其中內容改為:
numpy==1.23.0
scipy
open3d>=0.8
Pillow
tqdm

安裝Graspness相關依賴庫:
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
編譯安裝pointnet2:
cd pointnet2
python setup.py install
cd ..
編譯安裝knn:
cd knn
python setup.py install
cd ..
安裝graspnetAPI:
git clone https://github.com/graspnet/graspnetAPI.git
cd graspnetAPI
(修改2).需要修改graspnetAPI文件夾下的setup.py文件:


將sklearn替換為scikit-learn,指定numpy==1.23.0:

修改完成后繼續安裝:
pip install . -i https://pypi.tuna.tsinghua.edu.cn/simple
cd ..
如果到這里,上述步驟沒有問題,那么恭喜你,安裝配置成功。
二、代碼試運行
這里想運行graspnet-baseline文件夾中的示例文件:`demo.py`,需要將預訓練權重下載到graspnet-baseline文件夾中。前往官網Graspnet:https://github.com/graspnet/graspnet-baseline鼠標向下滾動至Training and Testing標題:

選擇任意的checkpoint.tar文件都可以,我這里使用的是checkpoint-rs.tar,因為實驗室的相機是Realsense的,如果你后續的實驗需要使用Kinect相機,更推薦你下載checkpoint-kn.tar文件。
使用百度網盤進行下載即可。
下載后將下載目錄下的文件復制到graspnet-baseline文件夾中即可:


完成后需要最后修改一個文件:
graspnet-baseline文件夾中的dataset/graspnet_dataset.py文件
將導包部分的代碼:
from torch._six import container_abcs
修改為:
import collections.abc as container_abcs
如下圖所示:

然后打開終端,激活虛擬環境,回到graspnet-baseline目錄下運行demo.py代碼:
cd ~/graspnet-baseline
conda activate graspnet
CUDA_VISIBLE_DEVICES=0 python demo.py --checkpoint_path checkpoint-rs.tar

效果圖如下:

至此,工成文畢!

浙公網安備 33010602011771號