Extract descriptor of single image using models trained on custom datasets

[JingjunYi]

Hello, thanks a lot for this valuable work. I tried your extract.py for image trieval during visual localization, it's great. But I want to train your SFRS on my own datasets and use the best model to extract descriptors of images. I tried to load pretrained models in model zoo, but it seems some parameters of base model and netvlad are missing. Only vgg16_netvlad.pth can be used for extraction, can you help me with this. How can I produce a new vgg16_netvlad.pth after training. I would be very grateful if you could answer my question.