Added support for Efficientnet backbones

gmberton / CosPlace

Official code for CVPR 2022 paper "Rethinking Visual Geo-localization for Large-Scale Applications"

MIT License

299 stars 58 forks source link

Closed lasuomela closed 1 year ago

lasuomela commented 1 year ago

Added support for some of the backbones from the EfficientNet family.

Test results:

train.py --backbone EfficientNet_B0
< test - #q: 1000; #db: 2805840 >: R@1: 63.6, R@5: 73.5, R@10: 77.0, R@20: 80.8

gmberton commented 1 year ago

Thanks for the PR! Just for the sake of completeness, could you add all 8 EfficientNet models?

lasuomela commented 1 year ago

Sure, there you go!

gmberton commented 1 year ago

PR merged, thank you!