videturfortuna / vehicle_reid_itsc2023

33 stars 5 forks source link

How to do rerank? #6

Open tryhardzz opened 11 months ago

tryhardzz commented 11 months ago

Hi, I found the re-ranking function in utils.py but how can I make use of that? I have already finished training and testing on veri776. Thank you for your time.

videturfortuna commented 11 months ago

Hi, I just updated teste.py. Just add flag --re_rank, for example:

python teste.py --path_weights ./weights/Veri776/MBR_4B_LAI/ --re_rank

mAP = 0.920961122545808, CMC1= 0.9803337454795837, CMC5= 0.9862931966781616