jwyang / fpn.pytorch

Pytorch implementation of Feature Pyramid Network (FPN) for Object Detection
MIT License
952 stars 221 forks source link

about rpn_loss_box is zero #46

Open ww451575464 opened 5 years ago

ww451575464 commented 5 years ago

i found rpn_loss_box has always been zero, and that because bbox_outside_weights is 0. maybe i should change positive_weights = 1.0 / num_examples negative_weights = 1.0 / num_examples to positive_weights = 1.0 / num_examples.item() negative_weights = 1.0 / num_examples.item() in anchor_target_layer_fpn.py line 136