Restormer 高解像度画像復元のための効率的なトランスフォーマー

Restormer コードのトレーニングとテスト実行の記録

記事とコードのアドレス

文章名称：Restormer: Efficient Transformer for High-Resolution Image Restoration（CVPR 2022）

コードアドレス: CVPR 2022 – Oral] 修復者

環境

3060 ラップトップ + WSL 22.04 + PyTorch 1.8.1 (利用不可)

インストール

1. リポジトリのクローンを作成し、フォルダーに入ります

   git clone https://github.com/swz30/Restormer.git
   cd Restormer
   

2. 新しい仮想環境を作成する

   conda create -n Restormer python=3.8
   conda activate Restormer
   

3. github ではrequirements.tet ファイルが提供されていないため、自分で作成します。

   touch requirements.txt
   vim requirements.txt
   

   以下の内容をコピーして貼り付け、右クリックします。

   matplotlib
   scikit-learn
   scikit-image
   opencv-python
   yacs
   joblib
   natsort
   h5py
   tqdm
   einops
   gdown
   addict
   future
   lmdb
   numpy
   pyyaml
   requests
   scipy
   tb-nightly
   yapf
   lpips
   

4. 依存関係をインストールする

   conda install pytorch=1.8 torchvision cudatoolkit=10.2 -c pytorch
   pip install requirements.txt
   

5. 基本をインストールする

   python setup.py develop --no_cuda_ext
   

   ここまでで環境は整いました。

データセットをダウンロードする

Google Cloud Disk からファイルをダウンロードするには、まず Golang をインストールし、次に Golang パッケージ管理ツール「go」を使用して「gdrive」をインストールする必要があります。

1. インストールする

   curl -O https://storage.googleapis.com/golang/go1.11.1.linux-amd64.tar.gz
   mkdir -p ~/installed
   tar -C ~/installed -xzf go1.11.1.linux-amd64.tar.gz
   mkdir -p ~/go
   

2. go to 環境変数を追加

   export GOPATH=$HOME/go
   export PATH=$PATH:$HOME/go/bin:$HOME/installed/go/bin
   

   具体的には、これら 2 つの環境変数の意味は次のとおりです。

   1. export GOPATH=$HOME/go: このコマンドは、「GOPATH」という環境変数を「$HOME/go"、ここで、"$ HOME " は現在のユーザーのホーム ディレクトリを表します。この環境変数は、コンパイラやその他のツールに、ソース コード、バイナリ、および Go 言語のその他の関連リソースを探す場所を指示します$。$
   2. export PATH=$PATH:$HOME/go/bin:$HOME/installed/go/bin: このコマンドは、システムの環境変数 "PATH" に 2 つのディレクトリ、つまり " HOME/go/bin" と "HOME/go/bin" を追加します。$HOME/go/bin"および$HOME/installed/go/bin"。これらのディレクトリには、"go" コマンドや "gofmt" コマンドなど、Go 言語に関連するいくつかの実行可能ファイルが含まれています。これらの$ディレクトリ$$"$ PATH" 環境変数を使用すると、フル パスを入力せずにコマンド ラインでこれらのコマンドを直接使用できます。

3. gドライブをインストールする

   go get github.com/prasmussen/gdrive
   

   GitHub から「gdrive」というリポジトリをダウンロードし、コンピュータにインストールします。

   注: 上記のコードは機能しない可能性があります。

   次に、手動インストールを通じて、次のコマンドを実行します。

   wget https://github.com/prasmussen/gdrive/releases/download/2.1.1/gdrive_2.1.1_linux_386.tar.gz
   tar -xvf gdrive_2.1.1_linux_386.tar.gz
   sudo mv gdrive /usr/local/bin/
   gdrive help
   

デモを実行する

訓練

テスト