Restormer代码训练和测试运行记录
文章及代码地址
文章名称:Restormer: Efficient Transformer for High-Resolution Image Restoration(CVPR 2022)
代码地址: CVPR 2022–Oral] Restormer
环境
3060 Laptop+WSL 22.04+PyTorch 1.8.1(无法使用)
安装
-
克隆仓库,并进入该文件夹
git clone https://github.com/swz30/Restormer.git cd Restormer
-
新建虚拟环境
conda create -n Restormer python=3.8 conda activate Restormer
-
因为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
-
安装依赖
conda install pytorch=1.8 torchvision cudatoolkit=10.2 -c pytorch pip install requirements.txt
-
安装basicsr
python setup.py develop --no_cuda_ext
至此环境已经准备完成。
下载数据集
从谷歌云盘下载文件,需要先安装Golang,然后使用Golang包管理工具“go”来安装“gdrive”。
-
安装 go
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
-
将go添加到环境变量中
export GOPATH=$HOME/go export PATH=$PATH:$HOME/go/bin:$HOME/installed/go/bin
具体来说,这两个环境变量的含义如下:
export GOPATH=$HOME/go
:这个命令将设置一个名为 “GOPATH” 的环境变量,它的值为 “ H O M E / g o " ,其中 " HOME/go",其中 " HOME/go",其中"HOME” 表示当前用户的home目录。这个环境变量告诉编译器和其他工具在哪里寻找 Go 语言的源代码、二进制文件和其他相关资源。export PATH=$PATH:$HOME/go/bin:$HOME/installed/go/bin
:这个命令将向系统的环境变量 “PATH” 中添加两个目录,分别为 “ H O M E / g o / b i n " 和 " HOME/go/bin" 和 " HOME/go/bin"和"HOME/installed/go/bin”。这些目录包含了一些与 Go 语言相关的可执行文件,例如 “go” 命令和 “gofmt” 命令等。通过将这些目录添加到 “PATH” 环境变量中,您可以在命令行中直接使用这些命令,而无需输入完整的路径。
-
安装 gdrive
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