MobileNet SSD : étude d'un détecteur d'objets embarquable entraîné sans images réelles - Archive ouverte HAL
Conference Papers Year : 2021

MobileNet SSD : étude d'un détecteur d'objets embarquable entraîné sans images réelles

Abstract

La détection d’objets en environnement industriel est un défi, d’une part à cause de l’absence d’images réelles pour l’apprentissage, et d’autre part à cause de la contrainte de temps réel requise pour l’algorithme, en général embarqué dans un dispositif mobile. Grâce aux données de conception 3D, il est cependant possible de créer une base d’images synthétiques annotées automatiquement pour entraîner un réseau de neurones artificiel, moyennant une performance limitée par l’écart de domaine avec la réalité. Dans cet article, nous étudions la performance d’un réseau de neurones à convolutions (CNN) pour la détection d’objets en temps réel et avec une faible empreinte mémoire : Single-Shot Detector (SSD), avec un réseau MobileNet comme extracteur de caractéristiques. Nous montrons qu’une stratégie d’augmentation adéquate permet d’entraîner SSD avec uniquement des images synthétiques photoréalistes pour détecter des objets industriels dans des images réelles. En particulier, sur la base de données T-LESS, SSD obtient une meilleure performance qu’un réseau Mask R-CNN, avec MobileNet-V2 et MobileNet-V3 comme extracteur de caractéristiques. Nos résultats montrent une amélioration de la précision d’environ 10% obtenue grâce à l’augmentation des données. Le code et les modèles entraînés seront disponibles en ligne.
Fichier principal
Vignette du fichier
ORASIS_JuliaCohen.pdf (5.66 Mo) Télécharger le fichier
Origin Files produced by the author(s)

Dates and versions

hal-03531390 , version 1 (18-01-2022)

Identifiers

  • HAL Id : hal-03531390 , version 1

Cite

Julia Cohen, Carlos F Crispim-Junior, Jean-Marc Chiappa, Laure Tougne. MobileNet SSD : étude d'un détecteur d'objets embarquable entraîné sans images réelles. ORASIS 2021, Centre National de la Recherche Scientifique [CNRS], Sep 2021, Saint Ferréol, France. ⟨hal-03531390⟩
736 View
1117 Download

Share

More