nabil_oh/SignsDetectionAI
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Utilisation seule de main pour entraîner puis lancer

Browse source
This commit is contained in:
yanis.bouarfa 2025-01-08 12:04:58 +01:00
parent 6da377d7c5
commit 9a0b597309
1 changed files with 11 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
main.py
View file

@ -1,9 +1,10 @@
import os import os
import subprocess
from src.pipeline import ObjectDetectionPipeline from src.pipeline import ObjectDetectionPipeline
from src.classifiers.bayesian import BayesianClassifier from src.classifiers.bayesian import BayesianClassifier
# Définissez le mode d'analyse ici : "plan" ou "page" # Définissez le mode d'analyse ici : "plan" ou "page"
analysis_mode = "page" analysis_mode = "plan"
if __name__ == "__main__": if __name__ == "__main__":
# Configuration basée sur le mode # Configuration basée sur le mode
@ -14,6 +15,15 @@ if __name__ == "__main__":
model_path = "models/bayesian_modelPAGE.pth" model_path = "models/bayesian_modelPAGE.pth"
image_path = "data/page.png" 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 # Chargement du modèle bayésien
print(f"Chargement du modèle bayésien depuis {model_path}") print(f"Chargement du modèle bayésien depuis {model_path}")
bayesian_model = BayesianClassifier(mode=analysis_mode) bayesian_model = BayesianClassifier(mode=analysis_mode)