search

Shiaoming / ALIKED

ALIKED: A Lighter Keypoint and Descriptor Extraction Network via Deformable Transformation
https://arxiv.org/pdf/2304.03608.pdf
BSD 3-Clause "New" or "Revised" License
190 stars 16 forks source link

Run ALIKED on CPU #3

Closed watatata closed 1 year ago

watatata commented 1 year ago

Thank you for releasing ALIKED.

I performed a demo on CPU as follows:

$ python demo_pair.py assets/st_pauls_cathedral --device cpu

Then the following error happened. 

    raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled

My PC (MacOS) does not support CUDA. Is ALIKED not yet CPU-compatible?

Shiaoming commented 1 year ago

I have not tested this code on MAC, but CPU mode should be supported on Ubuntu.