DiffBIR: Das intelligente Reparaturwerkzeug zur Verbesserung der Bildqualität

Allgemeine Einführung

DiffBIR (Blind Image Restoration with Generative Diffusion Prior) ist ein von der XPixelGroup entwickeltes Werkzeug zur blinden Bildwiederherstellung durch generative Diffusionsmodelle. DiffBIR verwendet ein fortschrittliches generatives Diffusionsmodell, um qualitativ hochwertige wiederhergestellte Bilder zu erzeugen, ohne sich auf ein bestimmtes Degradationsmodell zu verlassen. Das Projekt ist auf GitHub als Open-Source-Projekt verfügbar und bietet detaillierte Anleitungen und vortrainierte Modelle, um die Benutzer bei der Bildwiederherstellung zu unterstützen.

Online-Erfahrung: https://replicate.com/zsxkib/diffbir

Funktionsliste

• Blinde Bild-SuperauflösungVerbessert die Klarheit und Detailgenauigkeit von Bildern mit niedriger Auflösung.
• Wiederherstellung eines blinden GesichtsReparieren Sie qualitativ schlechte oder unscharfe Gesichtsbilder.
• Blinde BildentrauschungEntfernt Rauschen aus einem Bild und verbessert die Bildqualität.
• Generierung von DiffusionsmodellenImage restoration using generative diffusion modelling for multiple degradation scenarios.
• Pre-Training ModellBietet eine Vielzahl von vortrainierten Modellen zur Unterstützung verschiedener Arten von Bildwiederherstellungsaufgaben.
• Online-DemoEine Online-Demo wird zur Verfügung gestellt, damit die Benutzer die Ergebnisse der Bildwiederherstellung direkt erleben können.

Hilfe verwenden

Ablauf der Installation

1. Klon-Lager::

   git clone https://github.com/XPixelGroup/DiffBIR.git
cd DiffBIR

2. Installation von Abhängigkeiten::

   pip install -r anforderungen.txt

3. Download des vortrainierten ModellsDownload des trainierten Modells aufModellefinden Sie auf der Projektseite Links zum Herunterladen spezifischer Modelle.

Anweisungen für den Gebrauch

Bild-Restaurierung

1. Vorbereiten des Eingabebildes: Legen Sie das zu restaurierende Bild in denEingabenKatalog.
2. Ausführen von Inferenzskripten::

   python inference.py --input_dir inputs --output_dir outputs

Mit diesem Befehl wird eine Änderung in derEingabenDie Bilder im Verzeichnis werden wiederhergestellt, und die Ergebnisse werden in der DateiAusgängeKatalog.

Modellschulung

1. Vorbereiten des Datensatzes: Legen Sie den Trainingsdatensatz in denDatenKatalog, um sicherzustellen, dass das Datenformat den Anforderungen entspricht.
2. Führen Sie das Trainingsskript aus::

   python train_stage1.py --config configs/train_stage1.yaml
python train_stage2.py --config configs/train_stage2.yaml

Mit diesem Befehl werden die erste und zweite Stufe des Trainings nacheinander ausgeführt, um das endgültige Reparaturmodell zu erstellen.

Detaillierte Funktionsabläufe

1. Blinde Bild-Superauflösung::
   • Das Platzieren eines Bildes mit niedriger Auflösung auf demEingabenKatalog.
   • Führen Sie Inferenzskripte aus, um hochauflösende Bilder zu erzeugen.
   • SondeAusgängedas resultierende Bild im Katalog, um den Reparatureffekt zu bestätigen.
2. Wiederherstellung eines blinden Gesichts::
   • Das Platzieren eines unscharfen oder qualitativ schlechten Gesichtsbildes auf demEingabenKatalog.
   • Führen Sie das Inferenzskript aus, um das Gesichtsbild zu reparieren.
   • SondeAusgängedas resultierende Bild im Katalog, um den Reparatureffekt zu bestätigen.
3. Blinde Bildentrauschung::
   • Das Bild, das das Rauschen enthält, wird in dieEingabenKatalog.
   • Führen Sie das Inferenzskript aus, um Rauschen aus dem Bild zu entfernen.
   • SondeAusgängedas resultierende Bild im Katalog, um die Wirkung der Rauschunterdrückung zu bestätigen.

Diffbir 2.1 Integrationskit

Dieser Inhalt wurde vom Autor versteckt. Bitte geben Sie den Verifizierungscode ein, um den Inhalt zu sehen.

Captcha:

Bitte beachten Sie diese Website WeChat öffentliche Nummer, Antwort "CAPTCHA, eine Art Challenge-Response-Test (Computer)", erhalten Sie den Verifizierungscode. Suchen Sie in WeChat nach "Chef-KI-Austauschkreis"oder"Looks-AI" oder WeChat, indem Sie die rechte Seite des QR-Codes scannen, können Sie die öffentliche WeChat-Nummer dieser Website aufrufen.