HyperAIHyperAI

Command Palette

Search for a command to run...

SSD : Détecteur MultiBox à Simple Tir

Wei Liu Dragomir Anguelov Dumitru Erhan Christian Szegedy Scott Reed Cheng-Yang Fu Alexander C. Berg

Résumé

Nous présentons une méthode pour la détection d'objets dans des images en utilisant un seul réseau neuronal profond. Notre approche, nommée SSD (Single Shot MultiBox Detector), discrétise l'espace de sortie des boîtes englobantes en un ensemble de boîtes par défaut sur différentes rapports d'aspect et échelles pour chaque emplacement de la carte des caractéristiques. Au moment de la prédiction, le réseau génère des scores pour la présence de chaque catégorie d'objet dans chaque boîte par défaut et produit des ajustements à la boîte pour mieux correspondre à la forme de l'objet. De plus, le réseau combine les prédictions provenant de plusieurs cartes des caractéristiques avec différentes résolutions pour traiter naturellement les objets de diverses tailles. Notre modèle SSD est simple par rapport aux méthodes nécessitant des propositions d'objets car il élimine complètement la génération de propositions et l'étape ultérieure de rééchantillonnage des pixels ou des caractéristiques, et encapsule tous les calculs dans un seul réseau. Cela rend SSD facile à entraîner et simple à intégrer dans les systèmes qui nécessitent une composante de détection. Les résultats expérimentaux sur les jeux de données PASCAL VOC, MS COCO et ILSVRC confirment que SSD offre une précision comparable aux méthodes qui utilisent une étape supplémentaire de proposition d'objets et est beaucoup plus rapide, tout en fournissant un cadre unifié pour l'entraînement et l'inférence. Comparativement aux autres méthodes mono-étape, SSD offre une bien meilleure précision, même avec une taille d'image d'entrée plus petite. Pour une entrée de 300×300300 \times 300300×300, SSD atteint 72,1% mAP (mean Average Precision) sur le test VOC2007 à 58 images par seconde (FPS) sur une carte graphique Nvidia Titan X ; pour une entrée de 500×500500 \times 500500×500, SSD atteint 75,1% mAP, surpassant ainsi un modèle Faster R-CNN comparable à l'état de l'art. Le code source est disponible à l'adresse suivante : https://github.com/weiliu89/caffe/tree/ssd .


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