From 9a0b59730933ca54ff9c6c91389c70c5f16fad63 Mon Sep 17 00:00:00 2001 From: "yanis.bouarfa" Date: Wed, 8 Jan 2025 12:04:58 +0100 Subject: [PATCH] =?UTF-8?q?Utilisation=20seule=20de=20main=20pour=20entra?= =?UTF-8?q?=C3=AEner=20puis=20lancer?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index 11af68a..71fd102 100644 --- a/main.py +++ b/main.py @@ -1,9 +1,10 @@ import os +import subprocess from src.pipeline import ObjectDetectionPipeline from src.classifiers.bayesian import BayesianClassifier # Définissez le mode d'analyse ici : "plan" ou "page" -analysis_mode = "page" +analysis_mode = "plan" if __name__ == "__main__": # Configuration basée sur le mode @@ -14,6 +15,15 @@ if __name__ == "__main__": model_path = "models/bayesian_modelPAGE.pth" image_path = "data/page.png" + # Exécuter le script train.py avec le mode sélectionné + print(f"Lancement de l'entraînement avec le mode {analysis_mode}...") + try: + subprocess.run(["python", "train.py", "--mode", analysis_mode], check=True) + print("Entraînement terminé.") + except subprocess.CalledProcessError as e: + print(f"Erreur lors de l'entraînement : {e}") + exit(1) + # Chargement du modèle bayésien print(f"Chargement du modèle bayésien depuis {model_path}") bayesian_model = BayesianClassifier(mode=analysis_mode)