search

longcw / faster_rcnn_pytorch

Faster RCNN with PyTorch
MIT License
1.7k stars 466 forks source link

why regression loss divide by fg_cnt? #45

Open rimchang opened 6 years ago

rimchang commented 6 years ago

rpn_loss_box = F.smooth_l1_loss(rpn_bbox_pred, rpn_bbox_targets, size_average=False) / (fg_cnt + 1e-4) loss_box = F.smooth_l1_loss(bbox_pred, bbox_targets, size_average=False) / (fg_cnt + 1e-4)

I can see regression loss divided by fg_cnt why regression loss divide by fg_cnt? Is there anyone know? explan please