Add predicted objects to predict command

Closes #36 (closed)

dan/predict/__init__.py

@@ -106,4 +106,22 @@ def add_predict_parser(subcommands) -> None:
         help="String separators used to split text into lines.",
         required=False,
     )
+    parser.add_argument(
+        "--predict-objects",
+        action="store_true",
+        help="Whether to output objects when plotting attention maps.",
+        required=False,
+    )
+    parser.add_argument(
+        "--threshold-method",
+        help="Thresholding method.",
+        choices=["otsu", "simple"],
+        default="otsu",
+    )
+    parser.add_argument(
+        "--threshold-value",
+        help="Thresholding value.",
+        type=int,
+        default=0,
+    )
     parser.set_defaults(func=run)