Mmdetection: only one class, bug

[fengxiuyaun]

when num_classes=1, g = torch.abs(input.sigmoid().detach() - target)
RuntimeError: Expected object of type torch.cuda.FloatTensor but found type torch.cuda.LongTensor for argument #3 'other'

[ybai62868]

num_classes needs to be set to 2 (such as person + bkg ) and reg_class_agnostic in bbox_head should be set to "True".
I successfully get the person detector by mmdet. That's a very good work!

[fengxiuyaun]

thank you very much

[fengxiuyaun]

Excuse me!. Do you used retina_head to train only person???