文章目录

• • 1 概述
  • 2 快速使用
  • • 2.1 配置环境
    • 2.2 预处理数据
    • 2.3 QR Code实现超分辨率
    • 2.4 训练模型
  • 2.5 处理整个文件夹

1 概述

目前该项目还处于开发阶段，接下来的内容会展示一下目前的实现效果。

还有很多地方没有优化，会在后续的过程中逐步优化模型。

2 快速使用

2.1 配置环境

在当前目录下打开命令行窗口，运行以下代码

pip install -r environment.yml

2.2 预处理数据

在当前目录下打开命令行窗口，运行以下代码

python utils.py

处理原始数据，得到hr_img, lr_img, sr_img文件夹，其中hr_img里面的图片就是用于训练模型的数据

2.3 QR Code实现超分辨率

模型效果如图所示，其中，左、中、右分别为LR、SR、HR，其中LR图像为HR图像经过高斯噪声所得，SR图像为从LR图像超分得到，HR图像是使用工具生成的。

文件名：6941-v4.png

文件名：1050-v1.png

文件名：8239-v4.png

2.4 训练模型

在当前目录下打开命令行窗口，运行以下代码

python main.py --mode train

每一个epoch都会保存到model文件夹中

2.5 处理整个文件夹

在当前目录下打开命令行窗口，运行以下代码

python main.py --mode sr