Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
Real-ESRGAN_pytorch
Commits
abf11adb
Commit
abf11adb
authored
Jun 18, 2024
by
Rayyyyy
Browse files
Update README
parent
09cac702
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
15 additions
and
15 deletions
+15
-15
Contributors.md
Contributors.md
+2
-0
README.md
README.md
+13
-15
doc/pipeline.png
doc/pipeline.png
+0
-0
No files found.
Contributors.md
0 → 100644
View file @
abf11adb
# Contributors
None
\ No newline at end of file
README.md
View file @
abf11adb
...
@@ -6,7 +6,7 @@
...
@@ -6,7 +6,7 @@
生成网络: 采用ESRGAN的生成网络, 对于x4倍的超分辨, 网络完全按照ESRGAN的生成器执行;对x2和x1倍的超分辨, 网络先进行pixel-unshuffle(pixel-shuffl的反操作, pixel-shuffle可理解为通过压缩图像通道而对图像尺寸进行放大), 以降低图像分辨率为前提, 对图像通道数进行扩充, 然后将处理后的图像输入网络进行超分辨重建。
生成网络: 采用ESRGAN的生成网络, 对于x4倍的超分辨, 网络完全按照ESRGAN的生成器执行;对x2和x1倍的超分辨, 网络先进行pixel-unshuffle(pixel-shuffl的反操作, pixel-shuffle可理解为通过压缩图像通道而对图像尺寸进行放大), 以降低图像分辨率为前提, 对图像通道数进行扩充, 然后将处理后的图像输入网络进行超分辨重建。
<div
align=
center
>
<div
align=
center
>
<img
src=
"./doc/ESRGAN.png"
/>
<img
src=
"./doc/ESRGAN.png"
width=
2000
height=
500
/
>
</div>
</div>
对抗网络: 由于使用的复杂的构建数据集的方式, 所以需要使用更先进的判别器对生成图像进行判别。使用U-Net判别器可以在像素角度, 对单个生成的像素进行真假判断, 这能够在保证生成图像整体真实的情况下, 注重生成图像细节。
对抗网络: 由于使用的复杂的构建数据集的方式, 所以需要使用更先进的判别器对生成图像进行判别。使用U-Net判别器可以在像素角度, 对单个生成的像素进行真假判断, 这能够在保证生成图像整体真实的情况下, 注重生成图像细节。
...
@@ -17,14 +17,13 @@
...
@@ -17,14 +17,13 @@
## 算法原理
## 算法原理
通过使用更实用的退化过程合成训练对, 扩展强大的ESRGAN以恢复一般的真实世界LR图像。
通过使用更实用的退化过程合成训练对, 扩展强大的ESRGAN以恢复一般的真实世界LR图像。
<div
align=
center
>
<div
align=
center
>
<img
src=
"./doc/
High-order的
pipeline.png"
/>
<img
src=
"./doc/pipeline.png"
width=
2000
height=
750
/
>
</div>
</div>
## 环境配置
## 环境配置
-v 路径、docker_name和imageID根据实际情况修改
-v 路径、docker_name和imageID根据实际情况修改
### Docker(方法一)
### Docker(方法一)
```
bash
```
bash
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04.1-py38-latest
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04.1-py38-latest
...
@@ -36,7 +35,6 @@ python setup.py develop
...
@@ -36,7 +35,6 @@ python setup.py develop
```
```
### Dockerfile(方法二)
### Dockerfile(方法二)
```
bash
```
bash
cd
./docker
cd
./docker
...
@@ -214,12 +212,12 @@ train:
...
@@ -214,12 +212,12 @@ train:
1.
下载预训练模型到
`experiments/pretrained_models`
目录下:
1.
下载预训练模型到
`experiments/pretrained_models`
目录下:
-
*RealESRGAN_x4plus.pth*
:
+
*RealESRGAN_x4plus.pth*:
```bash
```bash
wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth -P experiments/pretrained_models
wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth -P experiments/pretrained_models
```
```
-
*RealESRGAN_x4plus_netD.pth*
:
+
*RealESRGAN_x4plus_netD.pth*:
```bash
```bash
wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.3/RealESRGAN_x4plus_netD.pth -P experiments/pretrained_models
wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.3/RealESRGAN_x4plus_netD.pth -P experiments/pretrained_models
```
```
...
@@ -301,7 +299,7 @@ bash run_train_multi.sh
...
@@ -301,7 +299,7 @@ bash run_train_multi.sh
## 推理
## 推理
下载预训练模型
[
RealESRGAN_x4plus.pth
](
https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth
)
, 将其放入 weights 文件夹下, 测试结果默认保存在results文件夹下。
下载预训练模型
[
RealESRGAN_x4plus.pth
](
https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth
)
, 将其放入 weights 文件夹下, 测试结果默认保存在results文件夹下。
```
bash
```
# 下载预训练模型
# 下载预训练模型
wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth -P weights
wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth -P weights
# 执行推理
# 执行推理
...
@@ -316,7 +314,7 @@ Model: RealESRGAN_x4plus
...
@@ -316,7 +314,7 @@ Model: RealESRGAN_x4plus
</div>
</div>
### 精度
### 精度
本项目基于ADE20K公开数据集的val数据进行效果验证,需先要对ADE20K val数据进行推理,得到推理后的结果(默认在results下)
.
本项目基于ADE20K公开数据集的val数据进行效果验证,需先要对ADE20K val数据进行推理,得到推理后的结果(默认在results下)
计算
`NIQE`
得分。
```
bash
```
bash
# 执行推理
# 执行推理
...
...
doc/
High-order的
pipeline.png
→
doc/pipeline.png
View file @
abf11adb
File moved
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment