megvii-research / BBN

The official PyTorch implementation of paper BBN: Bilateral-Branch Network with Cumulative Learning for Long-Tailed Visual Recognition
https://arxiv.org/abs/1912.02413
MIT License
658 stars 101 forks source link

feature compactness calculation #30

Open IssacCyj opened 3 years ago

IssacCyj commented 3 years ago

Hi,

thanks for the great work! I have a question about how to compute feature compactness as shown in Figure 6 in the original paper. What I did is to get resnet features (4096 dimensional tensor) before FC layer for each ground truth class, and compute the averaged distance with it's centroid. But the distances I got is generally greater than 50, even 100, as opposed to what's in the paper (<0.5).

Could you share how you computed the compactness for each class?

Thanks, Issac