FishYuLi / BalancedGroupSoftmax

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

Is there a Balanced Group Softmax implementation that works with detectron2 #18

Open jackkwok opened 3 years ago

jackkwok commented 3 years ago

If not, how can I go about implementing Balanced Group Softmax for detectron2?

Shanyaodedanshen commented 3 years ago

I also want to use BAGS in detectron2.

twangnh commented 3 years ago

hi, it should be easy to port group softmax to detectron2