HyperAIHyperAI

Command Palette

Search for a command to run...

ShadowMaskFormer : Embeddings de patchs augmentés par des masques pour l'élimination des ombres

Zhuohao Li Guoyang Xie* Guannan Jiang Zhichao Lu

Résumé

Le modèle Transformer est récemment apparu comme la référence de facto pour les tâches de vision par ordinateur et a également été appliqué avec succès à la suppression des ombres. Cependant, ces méthodes existantes reposent fortement sur des modifications complexes des mécanismes d'attention au sein des blocs Transformer, tout en utilisant un plongement de patch générique. En conséquence, cela entraîne souvent des conceptions architecturales complexes nécessitant des ressources de calcul supplémentaires. Dans ce travail, nous visons à explorer l'efficacité de l'intégration d'informations sur les ombres dans les premières étapes du traitement. À cette fin, nous proposons un cadre basé sur le Transformer avec un nouveau plongement de patch spécifiquement adapté à la suppression des ombres, baptisé ShadowMaskFormer. Plus précisément, nous présentons un plongement de patch augmenté par un masque simple et efficace pour intégrer les informations sur les ombres et favoriser l'accent mis par le modèle sur l'acquisition de connaissances pour les régions ombrées. Des expériences approfondies menées sur les jeux de données de référence ISTD, ISTD+ et SRD démontrent l'efficacité de notre méthode face aux approches de pointe actuelles tout en utilisant moins de paramètres du modèle. Notre implémentation est disponible à l'adresse suivante : https://github.com/lizhh268/ShadowMaskFormer.


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
ShadowMaskFormer : Embeddings de patchs augmentés par des masques pour l'élimination des ombres | Articles | HyperAI