Beachten Sie, dass diese Version nicht mit früheren Versionen kompatibel ist. Wenn Sie frühere Versionen verwenden möchten, lesen Sie bitte den Zweig old_version
.
Englisch |简体中文 GitHub | Gitee码云
Google Colab: GitHub-Link | Google Drive-Link
? Datensätze ⏬ Google Drive ⏬ 百度网盘 (提取码:basr)
? Trainingskurven in wandb
Befehle für Training und Tests
⚡ HOWTOs
BasicSR ( Basic Super Restoration ) ist eine Open-Source-Toolbox zur Bild- und Videowiederherstellung auf Basis von PyTorch, z. B. Superauflösung, Rauschunterdrückung, Unschärfe, Entfernung von JPEG-Artefakten usw.
(ESRGAN, EDVR, DNI, SFTGAN) (HandyView, HandyFigure, HandyCrawler, HandyWriting)
Wir bieten einfache Pipelines zum Trainieren/Testen/Inferenzieren von Modellen für einen schnellen Einstieg. Diese Pipelines/Befehle können nicht alle Fälle abdecken. Weitere Einzelheiten finden Sie in den folgenden Abschnitten.
GAN | |||||
---|---|---|---|---|---|
StilGAN2 | Zug | Schlussfolgerung | |||
Gesichtswiederherstellung | |||||
DFDNet | - | Schlussfolgerung | |||
Superauflösung | |||||
ESRGAN | TODO | TODO | SRGAN | TODO | TODO |
EDSR | TODO | TODO | SRResNet | TODO | TODO |
RCAN | TODO | TODO | |||
EDVR | TODO | TODO | DUF | - | TODO |
BasicVSR | TODO | TODO | TOF | - | TODO |
Entschärfen | |||||
DeblurGANv2 | - | TODO | |||
Denoise | |||||
RIDNet | - | TODO | CBDNet | - | TODO |
Klon-Repo
git clone https://github.com/xinntao/BasicSR.git
Abhängige Pakete installieren
cd BasicSR
pip install -r requirements.txt
Installieren Sie BasicSR
Bitte führen Sie die folgenden Befehle im BasicSR-Root-Pfad aus, um BasicSR zu installieren:
(Stellen Sie sicher, dass Ihre GCC-Version: gcc >= 5)
Wenn Sie die Cuda-Erweiterungen nicht benötigen:
dcn für EDVR
upfirdn2d und Fused_act für StyleGAN2
Bitte fügen Sie bei der Installation --no_cuda_ext
hinzu
python setup.py develop --no_cuda_ext
Wenn Sie das EDVR- und StyleGAN2-Modell verwenden, sind die oben genannten Cuda-Erweiterungen erforderlich.
python setup.py develop
Möglicherweise möchten Sie auch die CUDA-Pfade angeben:
CUDA_HOME=/usr/local/cuda
CUDNN_INCLUDE_DIR=/usr/local/cuda
CUDNN_LIB_DIR=/usr/local/cuda
python setup.py develop
Beachten Sie, dass BasicSR nur in Ubuntu getestet wurde und möglicherweise nicht für Windows geeignet ist. Sie können Windows WSL mit CUDA-Unterstützung ausprobieren :-) (Es ist jetzt nur für Insider-Builds mit Fast Ring verfügbar).
Bitte beachten Sie die Projektboards.
torch.utils.data.Dataset
-Klassen) finden Sie in Datasets.md. Die Designs und Konventionen der BasicSR-Codebasis finden Sie unter DesignConvention.md.
Die folgende Abbildung zeigt den Gesamtrahmen. Weitere Beschreibungen für jede Komponente:
Datensätze.md | Models.md | Config.md | Logging.md
Dieses Projekt wird unter der Apache 2.0-Lizenz veröffentlicht.
Weitere Einzelheiten zur Lizenz und Anerkennung finden Sie in LIZENZ.
Wenn BasicSR Ihrer Forschung oder Arbeit hilft, denken Sie bitte darüber nach, BasicSR zu zitieren.
Das Folgende ist eine BibTeX-Referenz. Der BibTeX-Eintrag erfordert die url
des LaTeX-Pakets.
@misc{wang2020basicsr,
author = {Xintao Wang and Ke Yu and Kelvin C.K. Chan and
Chao Dong and Chen Change Loy},
title = {BasicSR},
howpublished = { url {https://github.com/xinntao/BasicSR}},
year = {2020}
}
Xintao Wang, Ke Yu, Kelvin CK Chan, Chao Dong und Chen Change Loy. BasicSR. https://github.com/xinntao/BasicSR, 2020.
Wenn Sie Fragen haben, senden Sie bitte eine E-Mail an [email protected]
.