HyperAIHyperAI

Command Palette

Search for a command to run...

Modèle miniature et efficace pour la généralisation de la détection des contours

Soria Xavier ; Li Yachuan ; Rouhani Mohammad ; Sappa Angel D.

Résumé

La plupart des tâches de vision par ordinateur de haut niveau reposent sur des opérations d'image de bas niveau comme leurs processus initiaux. Des opérations telles que la détection de contours, l'amélioration d'images et la super-résolution fournissent les bases pour une analyse d'image plus avancée. Dans ce travail, nous abordons la détection de contours en considérant trois objectifs principaux : simplicité, efficacité et généralisation, car les modèles actuels de pointe (SOTA) pour la détection de contours ont augmenté en complexité pour améliorer leur précision. Pour atteindre ces objectifs, nous présentons Tiny and Efficient Edge Detector (TEED), un réseau neuronal convolutif léger avec seulement 58K paramètres, soit moins de 0,2 % des modèles de pointe. L'entraînement sur le jeu de données BIPED prend moins de 30 minutes, chaque époque nécessitant moins de 5 minutes. Notre modèle proposé est facile à entraîner et converge rapidement dès les premières époques, tout en produisant des cartes de contours nettes et de haute qualité. De plus, nous proposons un nouveau jeu de données pour tester la généralisation de la détection de contours, comprenant des échantillons d'images populaires utilisées dans la détection de contours et la segmentation d'images. Le code source est disponible sur https://github.com/xavysp/TEED.


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