GHM_Detection
GHM_Detection copied to clipboard
Is there a error in function `_expand_onehot_labels`?
Some code in function _expand_onehot_labels of https://github.com/open-mmlab/mmdetection/blob/master/mmdet/models/losses/ghm_loss.py is
if inds.numel() > 0:
bin_labels[inds, labels[inds]] = 1
but in you rep function, it is:
if inds.numel() > 0:
bin_labels[inds, labels[inds] - 1] = 1
Is there something wrong?