Pointnet++ 编译 tf_ops 遇到的问题及解决办法

最新推荐文章于 2025-02-28 21:33:48 发布

守护荆棘

最新推荐文章于 2025-02-28 21:33:48 发布

阅读量5.6k

点赞数 11

分类专栏：点云深度学习相关文章标签： pointnet++代码实现问题

本文链接：https://blog.youkuaiyun.com/SGL_LGS/article/details/100801061

版权

Pointnet++ 编译 tf_ops 遇到的问题及解决办法

pointnet++在pointnet 的基础上，提出了一个分层特征提取方法。其中涉及到采样（sampling layer），分组（grouping layer）和插值（interpolation layer）操作。再复现其代码时，最主要的问题就是集中在对上述三种操作的代码编译生成“xxxxxxx_so.so”文件。
在这里插入图片描述

1、修改编译文件，以“/samping/tf_sampling_compile.sh”为例

Github 上提供的源代码如下，有TF1.2和TF1.4两种版本

#/bin/bash
/usr/local/cuda-8.0/bin/nvcc tf_sampling_g.cu -o tf_sampling_g.cu.o -c -O2 -DGOOGLE_CUDA=1 -x cu -Xcompiler -fPIC

# TF1.2
g++ -std=c++11 tf_sampling.cpp tf_sampling_g.cu.o -o tf_sampling_so.so -shared -fPIC -I /usr/local/lib/python2.7/dist