Janspiry/Image-Super-Resolution-via-Iterative-Refinement
Image Super Resolution Via Iterative Refinement
Ce projet est un framework de deep learning pour la super-résolution d'image IA et la synthèse faciale. Il fournit un upscaler d'image par modèle de diffusion et un synthétiseur d'image faciale génératif capable de transformer des images basse résolution en sorties haute résolution en utilisant des poids de modèles pré-entraînés.
Le système utilise un raffinement de diffusion itératif et un échantillonnage guidé par basse résolution pour restaurer les détails fins et la netteté. Il prend en charge à la fois la génération d'image inconditionnelle, où les images sont créées à partir de zéro, et l'amélioration de résolution guidée pour une reconstruction faciale haute fidélité.
Le dépôt inclut un pipeline d'entraînement de modèle de diffusion avec entraînement distribué multi-GPU et initialisation des poids pré-entraînés. Cet environnement est pris en charge par le suivi des expériences de modèle, la journalisation des métriques externes et la reprise de modèle basée sur des points de contrôle (checkpoints).
