search

Parskatt / RoMa

[CVPR 2024] RoMa: Robust Dense Feature Matching; RoMa is the robust dense feature matcher capable of estimating pixel-dense warps and reliable certainties for almost any image pair.
https://parskatt.github.io/RoMa/
MIT License
622 stars 51 forks source link

Running the demo is not working #47

Closed geodesit closed 4 months ago

geodesit commented 5 months ago

I'm trying to run the demo_match.py inside a ubuntu20 docker image. I did the setup and I'm getting the error:

Screenshot from 2024-06-20 09-40-10

In what environment did you test it ? Do you have image with specific python libraries versions ?

alexstoken commented 4 months ago

I believe I ran into this recently, there's somewhere where the autocast breaks for CPU. I have a PR in the works, hopefully can finish it this weekend unless @Parskatt has it covered.

If you have a CUDA device the demo should work.

Parskatt commented 4 months ago

Yeah this is likely due to autocast. I have no time currently so PR would be appreciated 😀