chenhao2345 / ICE

Implementation for ICCV 2021 paper "ICE: Inter-instance Contrastive Encoding for Unsupervised Person Re-identification"
MIT License
44 stars 15 forks source link

Which part of the paper does "loss_ccl" in trainers.py correspond to,and how to improve the accuracy? #3

Open hzd7117 opened 2 years ago

hzd7117 commented 2 years ago

Hello, it's a great honor to read your excellent work, but when reproducing your paper code, I have the following questions: 1. Which part of the paper does "loss_ccl" in trainers.py correspond to? This loss function does not seem to improve the effect of the model. 2. At present, the highest accuracy of my reproduction is only 76.7%. How can I achieve the accuracy pointed out in the paper?

Thank you for your answer!

hzd7117 commented 2 years ago

In addition, add the experimental environment unbuntu16.04.6, pytorch1.7, and the graphics card is Tesla V100

chenhao2345 commented 2 years ago

Hi, loss_ccl corresponds to L_agnostic in Equation (5) of the paper.

Our results are obtained batch size and learning rate tuned with 4 GPUs. Please try to run our code with 4 GPUs, otherwise, you may have to tune batch size and learning rate for 1 GPU.

hzd7117 commented 2 years ago

thanks!