Make filter threshold an argument (#11)

Allow user to control the post-processing score filter threshold.

acrosome_counter/__main__.py

@@ -28,7 +28,7 @@ def main(args):
         cfg = build_cfg(
             is_training, args.batch_size, args.learning_rate, args.qty_iters,
         )
-        cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = .3
+        cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = args.threshold
         predictor = Predictor(cfg)
         predictor(dataset, plot=args.plot)
         predictor.export_xml()
@@ -45,6 +45,7 @@ def main(args):
     parser.add_argument('-bs', '--batch_size', default=1, type=int)
     parser.add_argument('-it', '--qty_iters', default=1, type=int)
     parser.add_argument('-lr', '--learning_rate', default=2.5E-4, type=float)
+    parser.add_argument('-t', '--threshold', default=.1, type=float)
     parser.add_argument('--train', action='store_true')
     parser.add_argument('--infer', action='store_true')
     parser.add_argument('--plot', action='store_true')