HyperAIHyperAI

Command Palette

Search for a command to run...

Modèle à deux étapes de segmentation et détection d'anomalies d'image par approche grossière-vers-finale

HyungWon Kim Odilbek Urmonov Rizwan Ali Shah

Résumé

Les approches existantes de détection et de segmentation des anomalies basées sur les réseaux de neurones convolutifs (CNN) sont soit trop sensibles, soit insuffisamment sensibles au bruit, ce qui entraîne une détection partielle des motifs anormaux lors de l’étape de test. Bien que ces méthodes puissent distinguer les images normales des images anormales, elles ne parviennent pas à localiser avec précision les anomalies présentes dans les images de test. Pour remédier à ce problème, nous proposons un modèle CNN à deux étapes pour la segmentation et la détection des anomalies de type grossier à fin, appelé TASAD (Two-stage Anomaly Segmentation and Detection). Dans les deux étapes de TASAD, nous entraînons notre modèle sur un mélange d’échantillons normaux et anormaux. Les images anormales sont obtenues en insérant des motifs pseudo-anormaux générés automatiquement à partir d’images sources d’anomalies. Nous utilisons une technique novatrice et sophistiquée d’insertion d’anomalies pour générer divers échantillons anormaux. Dans la première étape, nous concevons un modèle de segmentation d’anomalies grossières (CAS), qui prend une image entière en entrée. Dans la deuxième étape, nous entraînons un modèle de segmentation d’anomalies fines (FAS) sur des patches d’image. Le modèle FAS améliore les performances de détection et de segmentation en affinant les motifs anormaux partiellement détectés par le modèle CAS. Nous avons entraîné notre cadre sur le jeu de données MVTec et l’avons comparé aux méthodes de pointe (SOTA). L’architecture proposée conduit à une taille de modèle compacte — quatre fois plus petite que la méthode de pointe — tout en offrant une meilleure précision au niveau des pixels. TASAD peut également être appliqué aux méthodes de pointe pour améliorer davantage leurs performances en détection des anomalies. Nos expériences démontrent que, lorsqu’il est appliqué aux dernières méthodes de pointe, TASAD améliore la performance moyenne de précision (AP) des méthodes précédentes de 6,2 %. Pour assurer la reproductibilité des résultats, le code est disponible à l’adresse suivante : https://github.com/RizwanAliQau/tasad.git.


Créer de l'IA avec l'IA

De l'idée au lancement — accélérez votre développement IA avec le co-codage IA gratuit, un environnement prêt à l'emploi et le meilleur prix pour les GPU.

Codage assisté par IA
GPU prêts à l’emploi
Tarifs les plus avantageux

HyperAI Newsletters

Abonnez-vous à nos dernières mises à jour
Nous vous enverrons les dernières mises à jour de la semaine dans votre boîte de réception à neuf heures chaque lundi matin
Propulsé par MailChimp
Modèle à deux étapes de segmentation et détection d'anomalies d'image par approche grossière-vers-finale | Articles | HyperAI