search

marialeyvallina / generalized_contrastive_loss

MIT License
87 stars 14 forks source link

How to reproduce the result of RobotCar Seasons v2? #11

Closed MAX-OTW closed 1 year ago

MAX-OTW commented 1 year ago

Hi, @marialeyvallina @nicstrisc, I really appreciate for your great work. I'm trying to reproduce results of RobotCar Seasons v2. I have got the MSLS_resnet50_avg_480_GCL_predictions.npy.Could you please tell me how to get the results in the paper next? Could you please provide the subsequent code? Of course, detailed guidance would be better. I am always looking forward to your kind response. Best regards.

marialeyvallina commented 1 year ago

Hi @MAX-OTW You can reproduce the results on robotcar seasons by specifying the dataset as robotcarseasons while running the script, for example:

python3 extract_predictions.py --dataset robotcarseasons --root_dir /mydir/RobotCarSeasons/ --subset test --query_idx_file /mydir/RobotCarSeasons/test.json --map_idx_file /mydir/RobotCarSeasons/reference.json --model_file models/MSLS/MSLS_vgg16_avg_480_CL.pth --backbone vgg16 --pool avg