adhirajghosh / RPTM_reid

[WACV'23] The official code for "Relation Preserving Triplet Mining for Stabilising the Triplet Loss in Re-identification Systems"
https://openaccess.thecvf.com/content/WACV2023/papers/Ghosh_Relation_Preserving_Triplet_Mining_for_Stabilising_the_Triplet_Loss_In_WACV_2023_paper.pdf
MIT License
29 stars 3 forks source link

Only vehicle re-id #7

Open tjasmin111 opened 9 months ago

tjasmin111 commented 9 months ago

Hi. I'm only interested in vehicle re-id, not human. Can I only have that dataset and train the model?

adhirajghosh commented 3 weeks ago

Could you elaborate further? The code works for any re-identification or retrieval dataset. We have provided the dataset processing scripts for veri776, vehicleID and DukeMTMC, where only Duke is for person reID and the rest are for vehicle reID. Kindly look at these scripts for reference.

Keeping this issue open if you have further questions.