在点云中构建KNN近邻图（使用faiss库实现GPU快速计算KNN）_faiss实现knn

作者：知新_RL | 2024-02-28 16:09:39

踩

faiss实现knn

动机

点云识别中，有时为了利用图推理，需要利用点云构造拓扑图。构造的基本思路为：点云中的点位图的顶点，点云中每个点和距离该点最近的K个点之间连一条边。

实现

使用Faiss库中基于GPU的KNN快速计算。

细节待补充，参考GeoFormer中的实现

def init_knn(self):
    faiss_cfg = faiss.GpuIndexFlatConfig()
    faiss_cfg.useFloat16 = True
    faiss_cfg.device = 0

    self.geo_knn = faiss.GpuIndexFlatL2(faiss.StandardGpuResources(), 3, faiss_cfg)

1
2
3
4
5
6
7

应用

在GeoFormer中，利用该KNN Graph计算点的测地距离(Geodesic Distance)。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/知新_RL/article/detail/160041