【点云3D目标检测】跑通CIA-SSD过程中的一些报错集锦

前言

本文主要是介绍了在跑通CIA-SSD系列算法过程中遇到的一些bug,下面是对CIASSD算法的一个简单的介绍。
CIA-SSD其基本思想是校准单步目标检测中分类定位两个任务,提出Confident IoU-Aware Single-Stage object Detector (CIA-SSD)
第一个是Spatial-Semantic Feature Aggregation(SSFA)模块,为了准确预测目标框和分类置信度,自适应地融合低端spatial feature和高端抽象semantic feature。
第二个是IoU-aware confidence rectification模块,对置信度进一步校准(rectified),使其和定位精度更加一致。最后采用Distance-variant IoU-weighted NMS获得更平滑的回归并避免冗余预测。
GitHub链接:CIA-SSD。
下图是CIA-SSD与其他的点云3D目标检测算法做的比较:
【点云3D目标检测】跑通CIA-SSD过程中的一些报错集锦_第1张图片


1、没有THC/THCNumerics.cuh文件

报错代码:

In file included from /home/hzc/PythonProject/SE-SSD/spconv/src/spconv/reordering.cu:18:0:
/home/hzc/PythonProject/SE-SSD/spconv/include/spconv/reordering.cu.h:18:10: fatal error: THC/THCNumerics.cuh: No such file or directory
 #include <THC/THCNumerics.cuh>
          ^~~~~~~~~~~~~~~~~~~~~
compilation terminated.
src/spconv/CMakeFiles/spconv.dir/build.make:185: recipe for target 'src/spconv/CMakeFiles/spconv.dir/reordering.cu.o' failed
make[2]: *** [src/spconv/CMakeFiles/spconv.dir/reordering.cu.o] Error 1
make[2]: *** Waiting for unfinished jobs....

报错截图:
【点云3D目标检测】跑通CIA-SSD过程中的一些报错集锦_第2张图片

在编译过程中报错,fatal error: THC/THCNumerics.cuh: No such file or directory,这里我们可以直接将 #include 部分直接删去,或者升级pytorch的版本。

2. error: ‘AT_CHECK’ was not declared in this scope

报错内容:

src/iou3d.cpp:7:23: error: ‘AT_CHECK’ was not declared in this scope
 #define CHECK_CUDA(x) AT_CHECK(x.type().is_cuda(), #x, " must be a CUDAtensor ")

报错截图:
【点云3D目标检测】跑通CIA-SSD过程中的一些报错集锦_第3张图片
解决方法:参考链接error: ‘AT_CHECK’ was not declared in this scope,将AT_CHECK修改为AT_ASSERT。

3. No module named ‘det3d’

报错解决办法:再次进入cd det3d,重新编译python setup.py build develop即可。

4. ImportError: cannot import name ‘UnencryptedCookieSessionFactoryConfig’ from ‘pyramid.session’ (unknown location)

报错原因:直接命令行安装apexpip install apex,运行报错内容为:

Traceback (most recent call last):
  File "main.py", line 10, in <module>
    import apex
	"""
	"""
ImportError: cannot import name 'UnencryptedCookieSessionFactoryConfig' from 'pyramid.session' (unknown location)

解决办法:手动安装apex

git clone git://github.com/NVIDIA/apex
cd apex
pip install -v --no-cache-dir ./

5. python中自己编写的文件报错

(SESSD) hzc@cjailab:~/PythonProject/SE-SSD_fusion$ python tools/MX_fusion_train.py --config=examples/point_pillars/configs/MX_fusion_train_config.py --checkpoint=epoch_60.pth
True
Traceback (most recent call last):
  File "tools/MX_fusion_train.py", line 35, in <module>
    from fusion.preprocess_for_fusion import preprocess_for_fusion
ModuleNotFoundError: No module named 'fusion'

fusion是自己编写的python文件,但是在运行的时候出现了No module named 'fusion'报错,原因是没有在系统初始编译中注册该文件,解决办法如下图:
【点云3D目标检测】跑通CIA-SSD过程中的一些报错集锦_第4张图片
在训练函数中添加工程的路径:`sys.path.append(‘/home/xxxx/SE-SSD_fusion’)

6. KeyError: “XXX is not in the models registry“

一般出现此类错误,检查模块是否加入registry了,检查_init_文件,记得跑python setup install

你可能感兴趣的:(机器学习,计算机视觉,3D目标检测,目标检测,3d)