windows下面的SpinNet实现
论文:SpinNet: Learning a General Surface Descriptor for 3D Point Cloud Registration (CVPR 2021)
文章目录
前言
记录一下自己的跑代码的过程,防止忘记。跑代码的过程中记录下来的,遇到的一些问题的解决。
一、代码及论文地址
二、环境配置
电脑配置:笔记本gtx1650 cuda 10.0 pytorch 1.7
1.配置pointnet2_ops_lib
配置这个就是,去链接: link. https://github.com/erikwijmans/Pointnet2_PyTorch去把这个东西下载下来:
直接pip安装:
pip install -r requirements.txt
或者是以下等等
# 在命令提示符窗口中,使用cd进入到解压后的文件的路径下
# 输入
python setup.py install
pip install pointnet2_ops_lib/.
不知道为啥别人就这么弄没问题,我弄就一堆破问题。
问题1
这个东西就很烦,出了一大堆破问题,而且没多少人解决。
代码如下(示例):
问题2
或者说,spinnet提到的:
pip install "git+git://github.com/erikwijmans/Pointnet2_PyTorch.git#egg=pointnet2_ops&subdirectory=pointnet2_ops_lib"
这个报错是下载不了,然后看到一个解决方案是把git改成https
pip install "git+https://github.com/erikwijmans/Pointnet2_PyTorch.git#egg=pointnet2_ops&subdirectory=pointnet2_ops_lib"
然后还是不行
最终解决
安装visual studio 2017(其他版本都不行,原本安装的vs2019死活装不上去)
参考:链接: link.https://blog.csdn.net/goodgoodstudy___/article/details/121577651
我没咋看的懂,反正大概是说,vs2015不行,然后有个是说vs2019不行,emmm就解决了。
不知道为啥没得requirements.txt文件,但是问题不大,缺什么下什么剩下的环境配置好像没其他问题了。
2.3DMatch数据集下载与解压
github提供:链接: link.https://pan.baidu.com/share/init?surl=FB7IUbKAAlk7RVnB_AgwcQ(提取码:d1vn)
这里面是两个数据集KITTI和3DMatch,有点大,先下载3DMatch,
对我这种导师不配电脑的属实有一丢丢,不太友好了,准备换课题了。
里面就是这个:
接下来,:
1)合并3DMatch.tar.gz00 到 3DMatch.tar.gz10:
# 进入cmd命令提示符界面,cd进入当前目录,输入如下:
copy /b 3DMatch.tar.gz* 3DMatch.tar.gz
2)解压:
tar -zxvf tar.gz
windows下合并、解压tar.gz文件出现的问题
(不出意外的就是,我出意外了)
两个问题
1)合并之后的文件变大了,内存不够,合并之后应该是28G;
2)因为是用的win11,所以直接在3DMatch文件夹中用powershell去合并,解压。会出现合并出来的文件存在缺失的问题好像是,可以合并,但是后续的解压会报错。
3.代码的运行
这个问题就不知道为啥会出现了,我不理解:
pycharm运行:SpinNet-main\ThreeDMatch下的train
会报错:
1)dataset.py文件
这个部分会报错,27行这边,对应图片的第二行,把‘/’ 改成‘\’原因说不清楚,调试的时候看见training_data_files 显示的值在那个SpinNet-main/data/3DMatch/patches/train后面是‘\’在后面分割的时候把train//分给了要读取的文件名,导致ids在图中第四行变成 [] 。
ids = [file.split("\\")[-1] for file in training_data_files]
2)train.py 文件
这个就是源代码的 batch_size太大了,改成2就可以了。
总结
目前还在跑train,以后的以后再说。