HyperAIHyperAI

Command Palette

Search for a command to run...

Les sorties multiples suffisent pour le déflouage

Sidun Liu Peng Qiao Yong Dou

Résumé

La tâche de déflouage d’image est un problème mal posé, car elle admet une infinité de solutions réalisables pour une image floue donnée. Les approches modernes basées sur l’apprentissage profond négligent généralement l’apprentissage des noyaux de flou et recourent directement à un apprentissage supervisé end-to-end. Les jeux de données populaires de déflouage définissent l’étiquette comme l’une des solutions réalisables. Toutefois, nous estimons qu’il n’est pas raisonnable de spécifier directement une étiquette, en particulier lorsque celle-ci est échantillonnée selon une distribution aléatoire. Par conséquent, nous proposons que le réseau apprenne la distribution des solutions réalisables, et, à cette fin, concevons une nouvelle architecture à sorties multiples ainsi qu’une fonction de perte adaptée à l’apprentissage de distribution. Notre approche permet au modèle de produire plusieurs solutions réalisables afin d’approximer la distribution cible. Nous proposons également une nouvelle méthode de multiplexage de paramètres, qui réduit le nombre de paramètres et l’effort computationnel tout en améliorant les performances. Nous avons évalué notre méthode sur plusieurs modèles de déflouage d’image, y compris le modèle de pointe actuel NAFNet. L’amélioration du PSNR global optimal (en choisissant le meilleur score parmi plusieurs sorties pour chaque image de validation) dépasse les méthodes de comparaison de 0,11 à 0,18 dB. L’amélioration du PSNR du meilleur head unique (en sélectionnant le meilleur head parmi plusieurs sur l’ensemble de validation) dépasse les méthodes de comparaison de 0,04 à 0,08 dB. Le code source est disponible à l’adresse suivante : https://github.com/Liu-SD/multi-output-deblur.


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