Tous les projets
Intelligence Artificielle / R&D · 2025En cours

Pipeline de Vision par Ordinateur (EfficientNet-B0)

Pipeline complet de computer vision pour la classification d'images : du cycle de vie des données à l'optimisation d'hyperparamètres.

PythonPyTorchOpenCVEfficientNet-B0Docker

Contexte

Développement complet d'un pipeline de computer vision basé sur l'architecture EfficientNet-B0, pensé pour la classification d'images dans un contexte d'ingénierie de pointe et médical. L'objectif : un pipeline reproductible, conteneurisé et optimisé de bout en bout.

Cycle de vie des données

  • Ingestion et partitionnement de datasets complexes (train / validation / test stratifiés).
  • Augmentation de données à la volée via OpenCV (rotations, normalisation, recadrage adaptatif).
  • Versionnement des jeux de données et traçabilité des expériences.

Entraînement & Optimisation

  • Fine-tuning d'EfficientNet-B0 pré-entraîné, gel/dégel progressif des couches.
  • Optimisation d'hyperparamètres : learning rate scheduling, weight decay, taille de batch.
  • Suivi des métriques (accuracy, F1, matrice de confusion) et early stopping.

Industrialisation

Tout le pipeline est packagé dans une image Docker légère, garantissant un environnement d'entraînement et d'inférence reproductible, du poste local au serveur GPU.

Approche : un modèle n'a de valeur que s'il est mesurable, reproductible et déployable.