SPNet 代码运行成功记录

 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、再次运行代码。

无误!!!!

猜你喜欢

转载自blog.csdn.net/QYJ_WORKHARDING/article/details/131668348
今日推荐