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
316 stars 66 forks source link

The update of source feature in hybrid memory #23

Closed kuiran closed 3 years ago

kuiran commented 3 years ago

In your paper, "For the source-domain class centroids {w}, the k-th centroid wk is updated by the mean of the encoded features belonging to class k in the mini-batch." But where is the mean operation in your code?

yxgeee commented 3 years ago

In practice, I found that updating the centroids by the mean of the features is almost equal to updating the centroids by the corresponding features sequentially, achieving similar performances. So I use the latter one for brevity.

kuiran commented 3 years ago

Okay, thank you!