Fixing for onnx.

Author: datumbox

test/test_onnx.py

@@ -1,3 +1,4 @@
+from common_utils import set_rng_seed
 import io
 import torch
 from torchvision import ops
@@ -257,6 +258,8 @@ def get_features(self, images):
         return features
 
     def test_rpn(self):
+        set_rng_seed(0)
+
         class RPNModule(torch.nn.Module):
             def __init__(self_module):
                 super(RPNModule, self_module).__init__()

torchvision/models/detection/rpn.py

@@ -264,7 +264,8 @@ def filter_proposals(self, proposals, objectness, image_shapes, num_anchors_per_
             boxes, scores, lvl = boxes[keep], scores[keep], lvl[keep]
 
             # remove low scoring boxes
-            keep = torch.where(scores > self.score_thresh)[0]
+            # use >= for Backwards compatibility
+            keep = torch.where(scores >= self.score_thresh)[0]
             boxes, scores, lvl = boxes[keep], scores[keep], lvl[keep]
 
             # non-maximum suppression, independently done per level