[ADD] Adding a inference script to validate performance of models compared to each other.

.gitignore

@@ -30,4 +30,5 @@ lib64/
 #website
 generated_html/
 annotations/
-dataset/
+dataset/
+test/

inference.py

@@ -0,0 +1,64 @@
+import argparse
+import glob
+import os
+import matplotlib.pyplot as plt
+ 
+from ultralytics import YOLO
+
+# uv run predict.py  path_to_model.pt  path_to_images ./test
+def parse_args() -> argparse.Namespace:
+    parser = argparse.ArgumentParser(
+        description="Prédiction YOLO sur un dossier d'images avec sauvegarde des résultats annotés."
+    )
+
+    parser.add_argument(
+        "model_path",
+        type=str,
+        help="Chemin vers le modèle YOLO (.pt)",
+    )
+    parser.add_argument(
+        "images_path",
+        type=str,
+        help="Dossier contenant les images à traiter",
+    )
+    parser.add_argument(
+        "output_path",
+        type=str,
+        help="Dossier de sortie pour les images annotées",
+    )
+
+    return parser.parse_args()
+
+
+def predict(model_path:str, images_path:str, output_path:str):
+    model = YOLO(model_path)
+    images = glob.glob(os.path.join(images_path, "*.jpg"))
+    selection = images[:25]
+    print(selection)
+    results = model(selection)
+    name_model = model_path.split('/')
+
+    output_dir = output_path
+    model_dir = name_model[0]
+
+    if not os.path.isdir(output_dir):
+        os.mkdir(output_dir)
+
+    for i, result in enumerate(results):
+        boxes = result.boxes
+        im_array = result.plot()  # retourne un array numpy BGR avec les boxes dessinées
+
+        plt.figure(figsize=(10, 10))
+        plt.imshow(im_array[..., ::-1])  # conversion BGR -> RGB pour matplotlib
+        plt.axis("off")
+        plt.show()
+
+        result.save(filename=os.path.join(output_dir, model_dir, f"result_{i}.jpg"))  # sauvegarde avec nom unique
+
+def main() -> None:
+    args = parse_args()
+    predict(args.model_path, args.images_path, args.output_path)
+
+if __name__ == "__main__":
+    main()
+ 

requirement.txt

@@ -1,76 +1,128 @@
 anyio==4.14.2
-apturl==0.5.2
-blinker==1.4
-Brlapi==0.8.3
-certifi==2020.6.20
-chardet==4.0.0
-click==8.4.2
-colorama==0.4.4
-command-not-found==0.3
-cryptography==3.4.8
-cupshelpers==1.0
-dbus-python==1.2.18
-defer==1.0.6
-distro==1.7.0
-distro-info==1.1+ubuntu0.2
+argon2-cffi==25.1.0
+argon2-cffi-bindings==25.1.0
+arrow==1.4.0
+asttokens==3.0.2
+async-lru==2.3.0
+attrs==26.1.0
+babel==2.18.0
+beautifulsoup4==4.15.0
+bleach==6.4.0
+certifi==2026.7.22
+cffi==2.1.0
+charset-normalizer==3.4.9
+comm==0.2.3
+cuda-bindings==13.3.1
+cuda-pathfinder==1.6.0
+cuda-toolkit==13.0.3.0
+debugpy==1.8.21
+decorator==5.3.1
+defusedxml==0.7.1
 exceptiongroup==1.3.1
-filelock==3.31.1
+executing==2.2.1
+fastjsonschema==2.21.2
+filelock==3.32.0
+fqdn==1.5.1
 fsspec==2026.6.0
 h11==0.16.0
-hf==1.24.0
-hf-xet==1.5.2
 httpcore==1.0.9
-httplib2==0.20.2
 httpx==0.28.1
-huggingface_hub==1.24.0
-idna==3.3
-importlib-metadata==4.6.4
-jeepney==0.7.1
-keyring==23.5.0
-language-selector==0.1
-launchpadlib==1.10.16
-lazr.restfulclient==0.14.4
-lazr.uri==1.0.6
-louis==3.20.0
-macaroonbakery==1.3.1
-more-itertools==8.10.0
-netifaces==0.11.0
-oauthlib==3.2.0
-olefile==0.46
+idna==3.18
+ipykernel==7.3.0
+ipython==8.39.0
+ipywidgets==8.1.8
+isoduration==20.11.0
+jedi==0.20.0
+jinja2==3.1.6
+json5==0.15.0
+jsonpointer==3.1.1
+jsonschema==4.26.0
+jsonschema-specifications==2025.9.1
+jupyter==1.1.1
+jupyter-builder==1.1.1
+jupyter-client==8.9.1
+jupyter-console==6.6.3
+jupyter-core==5.9.1
+jupyter-events==0.12.1
+jupyter-lsp==2.3.1
+jupyter-server==2.20.0
+jupyter-server-terminals==0.5.4
+jupyterlab==4.6.2
+jupyterlab-pygments==0.3.0
+jupyterlab-server==2.28.0
+jupyterlab-widgets==3.0.16
+lark==1.3.1
+markupsafe==3.0.3
+matplotlib-inline==0.2.2
+mistune==3.3.4
+mpmath==1.3.0
+nbclient==0.11.0
+nbconvert==7.17.1
+nbformat==5.10.4
+nest-asyncio2==1.7.2
+networkx==3.4.2
+notebook==7.6.1
+notebook-shim==0.2.4
+numpy==2.2.6
+nvidia-cublas==13.1.1.3
+nvidia-cuda-cupti==13.0.85
+nvidia-cuda-nvrtc==13.0.88
+nvidia-cuda-runtime==13.0.96
+nvidia-cudnn-cu13==9.20.0.48
+nvidia-cufft==12.0.0.61
+nvidia-cufile==1.15.1.6
+nvidia-curand==10.4.0.35
+nvidia-cusolver==12.0.4.66
+nvidia-cusparse==12.6.3.3
+nvidia-cusparselt-cu13==0.8.1
+nvidia-nccl-cu13==2.29.7
+nvidia-nvjitlink==13.3.33
+nvidia-nvshmem-cu13==3.4.5
+nvidia-nvtx==13.0.85
+overrides==7.7.0
 packaging==26.2
-pexpect==4.8.0
-Pillow==9.0.1
-protobuf==3.12.4
+pandocfilters==1.5.1
+parso==0.8.7
+pexpect==4.9.0
+pillow==12.3.0
+platformdirs==4.11.0
+prometheus-client==0.25.0
+prompt-toolkit==3.0.52
+psutil==7.2.2
 ptyprocess==0.7.0
-pycairo==1.20.1
-pycups==2.0.1
-PyGObject==3.42.1
-PyJWT==2.3.0
-pymacaroons==0.13.0
-PyNaCl==1.5.0
-pyparsing==2.4.7
-pyRFC3339==1.1
-python-apt==2.4.0+ubuntu4.1
-python-dateutil==2.8.1
-python-debian==0.1.43+ubuntu1.1
-pytz==2022.1
-pyxdg==0.27
-PyYAML==5.4.1
-reportlab==3.6.8
-requests==2.25.1
-screen-resolution-extra==0.0.0
-SecretStorage==3.3.1
-six==1.16.0
-ssh-import-id==5.11
-systemd-python==234
-tqdm==4.69.0
-typing_extensions==4.16.0
-ubuntu-drivers-common==0.0.0
-ubuntu-pro-client==8001
-ufw==0.36.1
-unattended-upgrades==0.1
-urllib3==1.26.5
-wadllib==1.3.6
-xdg==5
-xkit==0.0.0
-zipp==1.0.0
+pure-eval==0.2.3
+pycparser==3.0
+pygments==2.20.0
+python-dateutil==2.9.0.post0
+python-json-logger==4.1.0
+pyyaml==6.0.3
+pyzmq==27.1.0
+referencing==0.37.0
+requests==2.34.2
+rfc3339-validator==0.1.4
+rfc3986-validator==0.1.1
+rfc3987-syntax==1.1.0
+rpds-py==0.30.0
+send2trash==2.1.0
+setuptools==83.0.0
+six==1.17.0
+soupsieve==2.9.1
+stack-data==0.6.3
+sympy==1.14.0
+terminado==0.18.1
+tinycss2==1.5.1
+tomli==2.4.1
+torch==2.13.0
+torchvision==0.28.0
+tornado==6.5.7
+traitlets==5.15.1
+triton==3.7.1
+typing-extensions==4.16.0
+tzdata==2026.3
+uri-template==1.3.0
+urllib3==2.7.0
+wcwidth==0.8.2
+webcolors==25.10.0
+webencodings==0.5.1
+websocket-client==1.9.0
+widgetsnbextension==4.0.15