Removing non-default threshold from reference scripts.

Author: datumbox

references/detection/train.py

@@ -94,7 +94,7 @@ def main(args):
     print("Creating model")
     kwargs = {}
     if "rcnn" in args.model:
-        kwargs["rpn_score_thresh"] = 0.5
+        kwargs["rpn_score_thresh"] = 0.0
     model = torchvision.models.detection.__dict__[args.model](num_classes=num_classes, pretrained=args.pretrained,
                                                               **kwargs)
     model.to(device)