Update detectron2/modeling/roi_heads/fast_rcnn.py

detectron2/modeling/roi_heads/fast_rcnn.py

@@ -685,7 +685,7 @@ class FastRCNNOutputLayers(nn.Module):
             # rpn_scores = [rpn_scores[0][rpn_scores[0] > 0.9]]
             # scores_bf_multiply = scores  #  as a backup
             #rpn_scores = [p.get('objectness_logits').sigmoid() for p in proposals]
-            scores = [((s) * torch.sigmoid(rpn_s[:, None])) ** 0.5 for s, rpn_s in zip(scores, rpn_scores)]
+            scores = [(torch.sigmoid(s) * torch.sigmoid(rpn_s[:, None])) ** 0.5 for s, rpn_s in zip(scores, rpn_scores)]
         return fast_rcnn_inference(
             boxes,
             scores,