cuda 实现sift gpu_【算法】FRNN：CUDA实现的GPU上的固定半径最近邻搜索

FRNN

A Fixed Nearest Neighbors Search implemented on CUDA with similar interface as pytorch3d.ops.knn_points.

Performance

Depenency

Tested with cuda 10.2, python 3.8 and pytorch 1.6.0 on ubuntu 18.04.

Should be also fine other versions of cuda/python/pytorch.

Install

git clone --recursive https://github.com/lxxue/FRNN.git
# install a prefix_sum routine first
cd FRNN/external/prefix_sum
python setup.py install

# install FRNN
cd ../../ # back to the {FRNN} directory
python setup.py install

Usage

For fixed nearest neighbor