Edit setup.py to be similar to setup.py of DKM repo

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

434 stars 33 forks source link

Edit setup.py to be similar to setup.py of DKM repo #10

Closed qkqhd222 closed 5 months ago

qkqhd222 commented 8 months ago

Hello. Thank you for sharing the excellent repo. Can you check this PR?

Changes : -Edit setup.py so that "from roma import roma_outdoor" after "pip install git+https://github.com/Parskatt/RoMa.git" works similar to https://github.com/Parskatt/DKM/blob/main/setup.py -Edit device argument in torch.linspace() so that gpu mem is allocated to each device instead of gpu0

Thank you.

Parskatt commented 5 months ago

Sorry for the late reply. Thanks for the contrib, looks good :)