Remove global device and add device as an input parameters

Parskatt / DKM

[CVPR 2023] DKM: Dense Kernelized Feature Matching for Geometry Estimation

https://parskatt.github.io/DKM/

Other

378 stars 28 forks source link

Closed ducha-aiki closed 1 year ago

ducha-aiki commented 1 year ago

Defaults are not changing. Now demo_match.py can run on torch.device('mps') device, if it such device is specified

ducha-aiki commented 1 year ago

On mps there are some problems with mapping the final output, however it is likely problem of pytorch mps backend, which still has lots of bugs.

dkmv3_warp_sacre_coeur

The cpu version works just fine on Apple device:

Parskatt commented 1 year ago

Looks good to me :)