There are some questions related to the cluster center in the code

yxgeee / SpCL

[NeurIPS-2020] Self-paced Contrastive Learning with Hybrid Memory for Domain Adaptive Object Re-ID.

https://yxgeee.github.io/projects/spcl

MIT License

318 stars 67 forks source link

There are some questions related to the cluster center in the code #38

Open WentaoTan opened 3 years ago

WentaoTan commented 3 years ago

I found that when calculating the loss in the dynamic memory bank, the cluster center feature is not used, or the instance-level feature is used. May I be the one who made a mistake？

yxgeee commented 3 years ago

Please refer to https://github.com/yxgeee/SpCL/blob/master/spcl/models/hm.py#L71

We calculate the cluster centers by dividing the feature sum by the number of samples.