Superpoint Network for Point Cloud Oversegmentation_ICCV_2021
一、在环境中安装需要的库
1、看代码标红的位置,需要安装相应的库,通常使用 pip install 库名 的方式
(鼠标指到标红波浪线处,会出现相应的建议,如果是需要安转的库名,就进行安装。)
2、有的导出的是编译的相应的.so文件
运行代码时,如果是导出.so文件出错,首先要保证你配置环境的时候,编译正确,有生成.so文件,那么就应该是路径导出的有问题,可进行修改。
二、运行代码,但是加载数据集的时候出问题
初始直接运行spnet训练的代码有误。发现没有读取到数据集(数量为0)
以为是自己的数据集有问题,实际上需要在superpoints_graph 里面先生成相关的数据,然后再spnet 代码中直接应用。
1、准备数据集
我使用的是s3dis数据集,首先下载数据集。数据集简介
需要选择数据集,然后填一个表格就可以下载数据集。
其中,indoor3d_sem_seg_hdf5_data是Pointnet、point-transformer等语义分割算法处理数据的方式。但是并不适用于SPNet的数据加载。(已验证,不是我想使用的数据)
2、处理数据集
需要经过superpoint_graph里的代码处理数据。
进入S3DIS.md中,我运行了这两步代码
生成了4个文件夹
发现生成的文件夹都与SPNet不适用。
红色框里的属性是需要使用的,但是其他三个文件夹都不包含所有框里的属性。
很愁。。。之后,董老师给了思路。曾经她复现时使用的数据是features_supervision文件夹下。
但是我没有生成(直接使用了老师的数据集)
之后发现是需要在superpoint_graph中运行下面的代码,来生成features_supervision文件夹。
(已成功运行)
(1)生成文件的过程,有遇到数据的错误:
是在处理Area_3/hallway_2/ hallway_2.txt时候出现了问题
查找区间范围 0-1193228,使用二分法估计在哪一个区间有问题。
我的出问题的区间是 926300-926350
如下,我将有问题的数据更改为 -9.160
(2)有遇到文件名称不相符的情况,将文件夹里的copy_Room_1.txt更改为copyRoom_1.txt
3、再次运行代码。
无误!!!!