Restormer Efficient Transformer for High-Resolution Image Restoration

最新推荐文章于 2025-03-11 19:27:52 发布

百年孤独百年

最新推荐文章于 2025-03-11 19:27:52 发布

阅读量774

点赞数

分类专栏：图像重建文章标签： transformer 深度学习人工智能 Restormer CVPR

本文链接：https://blog.csdn.net/qq_36693723/article/details/130631879

版权

图像重建专栏收录该内容

16 篇文章

订阅专栏

文章详细记录了Restormer模型的代码训练和测试过程，包括环境配置（3060Laptop+WSL2+PyTorch），依赖安装，数据集下载（使用Golang和gdrive从谷歌云盘下载），以及如何运行Demo进行训练和测试。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

文章目录

- Restormer代码训练和测试运行记录

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
```
具体来说，这两个环境变量的含义如下：
1. export GOPATH=$HOME/go：这个命令将设置一个名为 “GOPATH” 的环境变量，它的值为 “ $H OME / g o " ，其中 "$ HOME” 表示当前用户的home目录。这个环境变量告诉编译器和其他工具在哪里寻找 Go 语言的源代码、二进制文件和其他相关资源。
2. export PATH=$PATH:$HOME/go/bin:$HOME/installed/go/bin：这个命令将向系统的环境变量 “PATH” 中添加两个目录，分别为 “ $H OME / g o / 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