jw2yang commited on
Commit
227b589
1 Parent(s): fd5c34a

Update detectron2/modeling/roi_heads/fast_rcnn.py

Browse files
detectron2/modeling/roi_heads/fast_rcnn.py CHANGED
@@ -685,7 +685,7 @@ class FastRCNNOutputLayers(nn.Module):
685
  # rpn_scores = [rpn_scores[0][rpn_scores[0] > 0.9]]
686
  # scores_bf_multiply = scores # as a backup
687
  #rpn_scores = [p.get('objectness_logits').sigmoid() for p in proposals]
688
- scores = [((s) * torch.sigmoid(rpn_s[:, None])) ** 0.5 for s, rpn_s in zip(scores, rpn_scores)]
689
  return fast_rcnn_inference(
690
  boxes,
691
  scores,
 
685
  # rpn_scores = [rpn_scores[0][rpn_scores[0] > 0.9]]
686
  # scores_bf_multiply = scores # as a backup
687
  #rpn_scores = [p.get('objectness_logits').sigmoid() for p in proposals]
688
+ scores = [(torch.sigmoid(s) * torch.sigmoid(rpn_s[:, None])) ** 0.5 for s, rpn_s in zip(scores, rpn_scores)]
689
  return fast_rcnn_inference(
690
  boxes,
691
  scores,