Megvii-BaseDetection / DenseTeacher

DenseTeacher: Dense Pseudo-Label for Semi-supervised Object Detection
Apache License 2.0
120 stars 12 forks source link

Regression branch loss #7

Closed TerryYiDa closed 2 years ago

TerryYiDa commented 2 years ago

你好,代码中求无监督回归loss的时候,loss_deltas = (iou_loss * teacher_quality[b_mask]).mean(),但是teacher_quality有正有负,求出的loss_deltas也会为负,为什么?

ZRandomize commented 2 years ago

你好,teacher quality是一个logits(未过sigmoid),所以有正有负,这里代码可能是一个意外,我们的实验结果表明是否将teacher quality乘到iou_loss上不会对结果有什么影响,所以直接使用loss_deptas=iou_loss.mean()即可

Joker316701882 commented 2 years ago

Please reopen this issue if there is any further question.