HyperAIHyperAI

Command Palette

Search for a command to run...

ShadowDiffusion : Lorsque le Prior de Dégradation Rencontre le Modèle de Diffusion pour la Suppression d'Ombres

Lanqing Guo¹, Chong Wang¹, Wenhan Yang², Siyu Huang³, Yufei Wang¹, Hanspeter Pfister³, Bihan Wen¹

Résumé

Les méthodes d'apprentissage profond récentes ont obtenu des résultats prometteurs dans la suppression des ombres sur les images. Cependant, les images restaurées souffrent encore de défauts de bord insatisfaisants, en raison du manque d'intégration des modèles de dégradation et de la capacité limitée de modélisation. Notre travail aborde ces problèmes en proposant un cadre de diffusion unifié qui intègre à la fois les a priori sur l'image et les a priori de dégradation pour une suppression d'ombres très efficace. Plus précisément, nous proposons tout d'abord un modèle de dégradation des ombres, qui nous inspire pour construire un nouveau modèle de diffusion itératif, baptisé ShadowDiffusion (ShandowDiffusion). Ce modèle améliore considérablement la capacité du modèle à supprimer les ombres en affinant progressivement la sortie souhaitée grâce aux a priori de dégradation et aux a priori génératifs diffusifs, ce qui peut servir de nouvelle base solide pour la restauration d'images. De plus, ShadowDiffusion affine progressivement le masque d'ombre estimé comme tâche auxiliaire du générateur de diffusion, ce qui conduit à une génération d'images sans ombre plus précise et robuste. Nous menons des expériences approfondies sur trois jeux de données publics populaires, incluant ISTD, ISTD+ et SRD, pour valider l'efficacité de notre méthode. Comparativement aux méthodes les plus avancées actuellement disponibles, notre modèle réalise une amélioration significative en termes de PSNR (Peak Signal-to-Noise Ratio), passant de 31,69 dB à 34,73 dB sur le jeu de données SRD.


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
ShadowDiffusion : Lorsque le Prior de Dégradation Rencontre le Modèle de Diffusion pour la Suppression d'Ombres | Articles | HyperAI