Input prediction folder

closes #109 (closed) Add the possibility to process a folder instead of a single image in dan prediction.

dan/predict/__init__.py

@@ -14,13 +14,16 @@ def add_predict_parser(subcommands) -> None:
         description=__doc__,
         help=__doc__,
     )
-
     # Required arguments.
-    parser.add_argument(
+    image_or_folder_input = parser.add_mutually_exclusive_group(required=True)
+    image_or_folder_input.add_argument(
         "--image",
-        type=pathlib.Path,
         help="Path to the image to predict.",
-        required=True,
+    )
+    image_or_folder_input.add_argument(
+        "--image-dir",
+        type=pathlib.Path,
+        help="Path to the folder where the images to predict are stored.",
     )
     parser.add_argument(
         "--model",
@@ -48,6 +51,12 @@ def add_predict_parser(subcommands) -> None:
         required=True,
     )
     # Optional arguments.
+    parser.add_argument(
+        "--image-extension",
+        type=str,
+        help="The extension of the images in the folder.",
+        default=".jpg",
+    )
     parser.add_argument(
         "--scale",
         type=float,