請注意,此版本與先前的版本不相容。如果你想使用以前的版本,請參考old_version
分支。
English |簡體中文 GitHub | Gitee碼雲
Google Colab:GitHub 連結|谷歌雲端硬碟連結
?資料集 ⏬ Google Drive ⏬ 百度網盤 (擷取碼:basr)
? wandb 中的訓練曲線
訓練和測試命令
⚡ 操作方法
BasicSR( Basic S uper Restoration )是一個基於PyTorch的開源影像和視訊修復工具箱,如超解析度、去噪、去模糊、JPEG偽影去除等。
(ESRGAN、EDVR、DNI、SFTGAN) (HandyView、HandyFigure、HandyCrawler、HandyWriting)
我們提供簡單的管道來訓練/測試/推理模型以快速啟動。這些管道/命令無法涵蓋所有情況,更多詳細資訊將在以下部分中提供。
產生網路 | |||||
---|---|---|---|---|---|
樣式GAN2 | 火車 | 推理 | |||
臉部修復 | |||||
DFD網路 | - | 推理 | |||
超解析度 | |||||
ESRGAN | 待辦事項 | 待辦事項 | SRGAN | 待辦事項 | 待辦事項 |
EDSR | 待辦事項 | 待辦事項 | SRResNet | 待辦事項 | 待辦事項 |
遠端控制器網路 | 待辦事項 | 待辦事項 | |||
EDVR | 待辦事項 | 待辦事項 | 杜夫 | - | 待辦事項 |
基本振動時效 | 待辦事項 | 待辦事項 | 飛行時間 | - | 待辦事項 |
去模糊 | |||||
去模糊GANv2 | - | 待辦事項 | |||
去噪 | |||||
RID網路 | - | 待辦事項 | CBD網絡 | - | 待辦事項 |
克隆倉庫
git clone https://github.com/xinntao/BasicSR.git
安裝依賴套件
cd BasicSR
pip install -r requirements.txt
安裝基本SR
請在BasicSR根路徑中執行以下命令來安裝BasicSR:
(確保您的 GCC 版本:gcc >= 5)
如果您不需要 cuda 擴充:
EDVR 的DCN
StyleGAN2 的upfirdn2d和fused_act
安裝時請新增--no_cuda_ext
python setup.py develop --no_cuda_ext
如果使用EDVR和StyleGAN2模型,則需要上述cuda擴展。
python setup.py develop
您可能還想指定 CUDA 路徑:
CUDA_HOME=/usr/local/cuda
CUDNN_INCLUDE_DIR=/usr/local/cuda
CUDNN_LIB_DIR=/usr/local/cuda
python setup.py develop
請注意,BasicSR 僅在 Ubuntu 中進行了測試,可能不適用於 Windows。您可以嘗試支援 CUDA 的 Windows WSL :-)(現在僅適用於使用快速環的內部建置)。
請參閱項目板。
torch.utils.data.Dataset
類別)的描述位於 Datasets.md 中。 請參閱 DesignConvention.md 以了解 BasicSR 程式碼庫的設計和約定。
下圖展示了整體框架。每個組件的更多描述:
數據集.md |模型.md |配置.md |日誌記錄.md
該專案是在 Apache 2.0 許可證下發布的。
有關許可證和確認的更多詳細信息,請參閱許可證。
如果 BasicSR 對您的研究或工作有幫助,請考慮引用 BasicSR。
以下是 BibTeX 參考資料。 BibTeX 條目需要url
LaTeX 套件。
@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}
}
王新濤、餘柯、Kelvin CK Chan、Chao Dong 和 Chen Change Loy。基本SR。 https://github.com/xinntao/BasicSR,2020。
如果您有任何疑問,請發送電子郵件[email protected]
。