search

Ahmedest61 / VLAD-BuFF

VLAD-BuFF: Burst-aware Fast Feature Aggregation for Visual Place Recognition (ECCV 2024)
GNU General Public License v3.0
34 stars 3 forks source link

Releasing the model on torch.hub? #1

Open gmberton opened 1 month ago

gmberton commented 1 month ago

Hey @Ahmedest61, it was great meeting you at ECCV!

Are there any plans to release the trained model on torch.hub? It is quite simple to do and allows people to use your model with two lines of code, allowing more people to use your model and helping to spread your work! For example I did it for CosPlace, and the trained model can be automatically downloaded from anywhere without cloning the repo just like this

import torch
model = torch.hub.load("gmberton/cosplace", "get_trained_model", backbone="ResNet50", fc_output_dim=2048)

I'd like to add your model to the VPR-methods-evaluation repo and having the model on torch.hub would greatly help

Ahmedest61 commented 1 month ago

Hey @gmberton,

It was great meeting you at ECCV as well, and thank you for your suggestion!

Adding the trained model to Torch Hub is indeed on my to-do list, and I completely agree that it would make the model more accessible and easier to evaluate. I'll keep you posted once I make more progress on this!

Ahmedest61 commented 1 month ago

Hey @gmberton, I've added the trained model on Torch Hub.