search

cvg / Hierarchical-Localization

Visual localization made easy with hloc
Apache License 2.0
3.24k stars 599 forks source link

A stupid question, is it possible to get the local features for each input image? #382

Open lihaolin88 opened 8 months ago

lihaolin88 commented 8 months ago

I'm wondering if there is a function that can help to get the local features (with shape 256 W/8 H/8 ) for each input image; if there's no such a function, can you tell me where is the local features in the code, really appreciate for the help!

jytime commented 8 months ago

What do you mean by local features? Given the size you mentioned there, do you want a CxH_localxW_local feat map? For example, if you want features from NetVLAD, you can get all you need in https://github.com/cvg/Hierarchical-Localization/blob/e3e953f4db00c3b9b14951482349d5ddd9424452/hloc/extractors/netvlad.py#L126