LightGlue 是一種用於局部特徵匹配的圖神經網絡,它會內省其置信度:
- 如果所有預測都已準備好,則儘早停止;
- 刪除被認為不匹配的點以節省計算量。
LightGlue 是一種輕量級特徵匹配器,具有高精度和自適應修剪技術,無論是在網絡的寬度還是深度上,都可以實現極快的推理。它將每個圖像的一組關鍵點和描述符作為輸入,並返回它們之間對應點的索引。
項目團隊已發布了具有SuperPoint和DISK局部特徵的LightGlue 預訓練權重。
LightGlue 可以調整每個圖像對的深度(層數)和寬度(關鍵點數量),對準確性的影響最小。
depth_confidence
:控制提前停止,改善運行時間。推薦值:0.95。默認值:-1(關閉)width_confidence
:控制迭代特徵刪除,提高運行時間。推薦值:0.99。默認值:-1(關閉)flash
:啟用FlashAttention。顯著提高運行時間並減少內存消耗,而不會對準確性產生任何影響,但需要FlashAttention或torch >= 2.0
#LightGlue首頁文檔和下載 #光速局部特徵匹配 #程式开发