CUDA accelerated SIFT in Python
This library is a wrapper around PopSift to compute SIFT keypoints and descriptors on the GPU using CUDA. It's written to be a drop-in replacement for existing OpenCV functions such as cv2.FeatureDetector_create('SIFT')
and cv2.DescriptorExtractor_create('SIFT')
.
Requirements:
- CUDA Toolkit >= 7
- CMake >= 3.14
- A C++11 capable compiler (g++ 5.4.0 works fine)
# git clone --recurse-submodules https://github.com/uav4geo/pypopsift # cd pypopsift && mkdir build && cd build # cmake .. # make -j8
To install the Python package:
# cd pypopsift # pip install .
python example.py
Mozilla Public License 2.0
@inproceedings{Griwodz2018Popsift, author = {Griwodz, Carsten and Calvet, Lilian and Halvorsen, P{\aa}l}, title = {Popsift: A Faithful SIFT Implementation for Real-time Applications}, booktitle = {Proceedings of the 9th {ACM} Multimedia Systems Conference}, series = {MMSys '18}, year = {2018}, isbn = {978-1-4503-5192-8}, location = {Amsterdam, Netherlands}, pages = {415--420}, numpages = {6}, doi = {10.1145/3204949.3208136}, acmid = {3208136}, publisher = {ACM}, address = {New York, NY, USA}, }