このバージョンは以前のバージョンと互換性がないことに注意してください。以前のものを使用したい場合は、 old_version
ブランチを参照してください。
English |简体中文 GitHub |ギティーコード云
Google Colab: GitHub リンク| Googleドライブリンク
?データセット ⏬ Google ドライブ ⏬ 百度网盘 (提取码:basr)
?ワンドブのトレーニングカーブ
トレーニングとテスト用のコマンド
⚡ ハウツー
BasicSR ( Basic Super Restoration ) は、超解像度、ノイズ除去、ブラー除去、JPEG アーチファクト除去などの、PyTorch に基づくオープンソースの画像およびビデオ復元ツールボックスです。
(ESRGAN、EDVR、DNI、SFTGAN) (HandyView、HandyFigure、HandyCrawler、HandyWriting)
迅速に開始できるように、モデルをトレーニング/テスト/推論するためのシンプルなパイプラインを提供します。これらのパイプライン/コマンドはすべてのケースをカバーできるわけではありません。詳細については次のセクションで説明します。
ガン | |||||
---|---|---|---|---|---|
スタイルGAN2 | 電車 | 推論 | |||
顔の修復 | |||||
DFDネット | - | 推論 | |||
超解像 | |||||
エスガン | TODO | TODO | サーガン | TODO | TODO |
EDSR | TODO | TODO | SRResNet | TODO | TODO |
RCAN | TODO | TODO | |||
EDVR | TODO | TODO | ダフ | - | TODO |
基本VSR | TODO | TODO | TOF | - | TODO |
ブレを除去する | |||||
DeblurGANv2 | - | TODO | |||
ノイズ除去 | |||||
RIDネット | - | TODO | CBDネット | - | TODO |
リポジトリのクローンを作成する
git clone https://github.com/xinntao/BasicSR.git
依存パッケージをインストールする
cd BasicSR
pip install -r requirements.txt
BasicSR のインストール
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 を試してみてください:-) (現在、高速リングを使用した Insider ビルドでのみ利用可能です)。
プロジェクトボードをご覧ください。
torch.utils.data.Dataset
クラス) の説明は Datasets.md にあります。 BasicSR コードベースの設計と規約については、DesignConvention.md を参照してください。
下図は全体の枠組みを示しています。各コンポーネントの詳細な説明:
データセット.md |モデル.md |構成.md |ロギング.md
このプロジェクトは、Apache 2.0 ライセンスに基づいてリリースされています。
ライセンスと承認の詳細については、「LICENSE」を参照してください。
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}
}
Xintao Wang、Ke Yu、Kelvin CK Chan、Chao Dong、Chen Change Loy。基本SR。 https://github.com/xinntao/BasicSR、2020 年。
ご質問がある場合は、 [email protected]
電子メールでお問い合わせください。