超分辨率代码运行

本篇文章将详细讲述新手小白初次运行超分辨率代码,以亚像素卷积(ESPCN)这篇文章为例。
(1)首先从github上下载ESPCN的代码(代码地址:https://github.com/yjn870/ESPCN-pytorch)下载完成后直接解压就ok了。
(2)将代码导入到pycharm中:打开pycharm,点击File——open,找到ESPCN代码文件夹,之后就项目就会导入,如下图所示:
在这里插入图片描述
(3)更改项目环境:点击File—settings—Project—python Interpreter
在这里插入图片描述
在这里插入图片描述
点击show All
选择之前安装cuda ,tourch的环境,这样就可以把环境配置好啦,之后就可以运行代码了
(4)打开代码中README.md文件,此文件是原作者对代码整体的解释,教你如何训练,测试代码以及代码所需要的环境。
首先看训练参数的设置
在这里插入图片描述
91-image_x3.h5是我们训练的数据集,已经将图片转换成.h5文件,其中X3中的3是3倍采样因子,Set5_x3.h5是验证数据集,他将两个数据集都放在BLAH_BLAH文件夹下,所以我们需要先创建一个存放数据集的文件夹BLAH_BLAH(名称随便),之后再BLAH_BLAH中创建一个outputs文件夹来存放输出的文件。之后将数据集下载到对应的目录。如下图所示
在这里插入图片描述
(5)先运行一下train.py ,会出现报错,不要慌,是因为你的训练参数还没有传入到网络中。根据下图
在这里插入图片描述
红色方框的内容是我们刚才运行的文件代码train.py(因为我的界面有两个项目,所以有两个train,不能同时使用,所以改了名称为train1),
点击下三角,点击Edit config,显示如下界面
在这里插入图片描述
找到Parameters

复制训练参数
–train-file “BLAH_BLAH/91-image_x3.h5”
–eval-file “BLAH_BLAH/Set5_x3.h5”
–outputs-dir “BLAH_BLAH/outputs”
–scale 3
–lr 1e-3
–batch-size 16
–num-epochs 200
–num-workers 8
–seed 123
到空格中,点击边上缩放箭头,修改对应数据集文件夹位置 及参数
在这里插入图片描述
如果出现页面文件太小等之类的错误,尝试将batch size,或者num workers设置小一点,因为电脑配置问题,我设置为16和4,迭代次数设置为10,如上图。
参数设置好之后,再次运行train.py 文件,代码就会跑起来啦
在这里插入图片描述
每一代的结果都会存放再outputs中。
以上就是整个训练的过程。

之后便是测试
先下载预训练权重文件到对应文件夹中,
在这里插入图片描述
之后再运行test.py,和之前运行train.py过程一样,
在这里插入图片描述
测试参数
如下所示:
在这里插入图片描述
其中从data中选择一张图片进行测试
之后运行test.py
结果如下
在这里插入图片描述
在这里插入图片描述
会额外出现两张结果图片
到此测试程序完成!!!!!!!!

  • 8
    点赞
  • 29
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值