search

FishYuLi / BalancedGroupSoftmax

CVPR 2020 oral paper: Overcoming Classifier Imbalance for Long-tail Object Detection with Balanced Group Softmax.
Apache License 2.0
351 stars 64 forks source link

Which file includes GroupSoftmax? #6

Open ShaneShen opened 4 years ago

ShaneShen commented 4 years ago

Hi, I found this work very interesting. However, I still cannot find the loss function that includes GroupSoftmax. Could you please tell me which file I should read?

FishYuLi commented 4 years ago

@ShaneShen Hi, please read mmdet/models/bbox_heads/gs_bbox_head_with0.py.