超分重建网络——DASR

视频图像增强


1.前言

为了督促自己认真学习,碧波自己把最近学的东西,写成博客/(ㄒoㄒ)/~~
在这里插入图片描述


2.DASR

2.1 论文和代码

想深刻学习这个网络的提出思路,推荐大家看下面这个博主的连接,他比我讲的好,所以我就不过多解释了。
论文解读:https://blog.csdn.net/weixin_43972154/article/details/119327182
至于学习代码的话,根据作者的操作步骤一步步来就ok了,学习新网络,最重要的是复现代码。
代码连接:https://github.com/The-Learning-And-Vision-Atelier-LAVA/DASR

3.代码复现

3.1 准备环境

由于作者使用的pytorch的环境比较老,所以用高版本的pytorch总是会出现难以想象的问题,为了节约时间,我建议大家采用虚拟环境的方式。操作起来非常简单,三步就OK了:

  1. 创建虚拟环境conda create -n pytorch1.1 python=3.6一路遇到选择选y
  2. 激活创建的虚拟环境activate pytorch1.1
  3. 在激活的虚拟环境下安装需要用到的包pip install -i https://pypi.douban.com/simple -r requirement.txt(记得将会用到的包放到个txt文件中)

3.2 数据准备

在这里插入图片描述
下载这两个数据放到一个文件夹里,为了避免少出错,我们把文件夹命名为DF2K

  1. 准备DIV2K

DIV2K数据集下载连接:https://data.vision.ee.ethz.ch/cvl/DIV2K/
这个连接里面有很多数据集,我当时没有搞清楚要下载那些数据集,差点全部下载了,注意只需要下载这两幅数据集,至于为什么,可以看我上面提供的论文解读。
在这里插入图片描述

  1. 准备Flickr2K

Flickr2K数据集下载连接:https://cv.snu.ac.kr/research/EDSR/Flickr2K.tar

  1. 合并数据

数据下载好了之后,放入文件夹DF2K里面,大概是这样子。
在这里插入图片描述
细心的网络可能会问,我为什么只有900张图片,是因为我的电脑显存不够,我只用了DIV2K数据集做训练。
在这里插入图片描述

3.3 开始训练

接下来就是开始训练了,训练代码,如果是用服务器的话直接run main.sh就可以,没有条件的可以使用以下命令:

python main.py --dir_data=./dataset/ --model=blindsr --scale=4 --blur_type=aniso_gaussian --noise=25.0 --lambda_min=0.2 --lambda_max=4.0

训练过程大概是这样的
在这里插入图片描述


注意出现以下问题:
在这里插入图片描述
在这里插入图片描述


3.4 测试数据

在这里插入图片描述

  1. 原论文准备了四种测试数据集,下面说一下数据的[下载连接]。(https://github.com/XPixelGroup/BasicSR/blob/a19aac61b277f64be050cef7fe578a121d944a0e/docs/Datasets.md)
    在这里插入图片描述
  2. 我测试的时候只用了Set14数据集,我用了这两个文件夹。

在这里插入图片描述

  1. Set14HR图像和LR图像,放到benchmark文件夹里,数据的存放文件格式是这样子的。

在这里插入图片描述


3.5 开启测试

在这里插入图片描述
下面就是开启测试了,测试命令如下:

 python test.py --test_only  --dir_data=./dataset/ --data_test=Set14 --model=blindsr  --scale=4 --resume=114 --blur_type=aniso_gaussian --noise=10.0 --theta=0.0  --lambda_1=0.2 --lambda_2=4.0 --save_results=True

在这里插入图片描述
以下问题:
在这里插入图片描述
另外,也可以使用官方提供的权重,快速测试。
在这里插入图片描述

3.6 测试结果

这里要提一下,训练过程结果保存在experiment文件下面,你用什么模式,就在什么文件夹里。
在这里插入图片描述
这里展示以下,我训练的结果,由于我只训练了114epoch,所以效果不好,勉为其难的展示一下吧。
在这里插入图片描述
在这里插入图片描述

在这里插入图片描述


4. 我的代码

下面提供以下,我的代码连接,比较懒的同学,可以直接下载我的代码,数据啥的我都已经下载好了,就不用麻烦挨个下载数据集了,直接使用。
链接:https://pan.baidu.com/s/1f_Gq-pvMyfth-FebiYpx7A
提取码:fasf
–来自百度网盘超级会员V5的分享

  • 8
    点赞
  • 28
    收藏
    觉得还不错? 一键收藏
  • 14
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 14
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值