图像去噪——AdaFM模型推理测试(详细图文教程)

news/2024/7/21 4:52:12 标签: python, 图像处理, pytorch

在这里插入图片描述

目录

  • 一、 源码包下载
  • 二、测试集和模型权重准备
    • 2.1 测试集
    • 2.2 模型权重
  • 三、参数修改
  • 四、推理测试
    • 4.1 固定插值系数图像测试
      • 4.1.1 测试结果保存
      • 4.1.2 测试结果展示
        • 4.1.2.1 场景1
        • 4.1.2.2 场景2
        • 4.1.2.3 场景3
        • 4.1.2.4 场景4
        • 4.1.2.5 场景5
    • 4.2 等差改变插值系数图像测试
      • 4.2.1 参数修改
      • 4.2.2 测试
      • 4.2.3 测试结果保存
      • 4.2.4 测试结果展示
        • 4.2.4.1 场景1
        • 4.2.4.2 场景2
        • 4.2.4.3 场景3
        • 4.2.4.4 场景4
        • 4.2.4.5 场景5
  • 五、推理速度
    • 5.1 GPU推理
    • 5.2 CPU推理
      • 5.2.1 CPU推理代码修改
  • 六、总结

一、 源码包下载

官网代码下载链接为:官网源码包

我提供的源码包下载链接:网盘源码包,提取码为:vz48

该篇论文方法代码我没修改太多,学者下载官网或者我提供的源码包都行。

论文地址:论文

源码包下载并解压后的文件列表如下:
在这里插入图片描述

二、测试集和模型权重准备

2.1 测试集

测试集图片放到根目录下的datasets文件夹下,如下:

在这里插入图片描述

2.2 模型权重

模型权重存放在根目录下的experiments文件夹下,模型权重大小为5.4M,如下:

在这里插入图片描述

三、参数修改

配置文件参数修改如下:

在这里插入图片描述

四、推理测试

4.1 固定插值系数图像测试

官网提供的源码包中测试需要在终端中输入命令测试,命令为:

python">cd codes
python test.py -opt options/test/test.json

我提供的源码包测试代码为,直接运行test.py脚步即可,不用在终端输入命令,如下:

在这里插入图片描述

4.1.1 测试结果保存

运行test.py脚本后测试结果会自动保存到根目录下的results路径下,如下:

在这里插入图片描述

4.1.2 测试结果展示

4.1.2.1 场景1

在这里插入图片描述

4.1.2.2 场景2

在这里插入图片描述

4.1.2.3 场景3

在这里插入图片描述

4.1.2.4 场景4

在这里插入图片描述

4.1.2.5 场景5

在这里插入图片描述

4.2 等差改变插值系数图像测试

在调制测试中,连续调节插值系数λ可以逐渐改变输出效果。如果输入图像是固定的,那么随着λ的增加,输出图像会变得更清晰或更平滑。

等差按照固定步长调整插值系数测试的方法在脚本interpolate.py中。

4.2.1 参数修改

测试集路径和模型权重路径同上,这里主要修改插值系数步长,如下:

在这里插入图片描述

上面修改的步长参数最终传值,主要传到位置如下:

在这里插入图片描述

4.2.2 测试

同样,官网下载的源码包需要在终端中输入下面命令运行:

python">cd codes
python interpolate.py -opt options/test/test.json

我提供的源码包直接运行脚本interpolate.py即可,如下:

在这里插入图片描述

4.2.3 测试结果保存

运行上面脚本后,会在根目录results文件夹中生成以各张图片名为名字的文件夹,在各个图片名对应的文件夹中又保存了不同插值系数处理的结果图,如下:

在这里插入图片描述

4.2.4 测试结果展示

4.2.4.1 场景1

在这里插入图片描述

4.2.4.2 场景2

在这里插入图片描述

4.2.4.3 场景3

在这里插入图片描述

4.2.4.4 场景4

在这里插入图片描述

4.2.4.5 场景5

在这里插入图片描述

五、推理速度

5.1 GPU推理

测试图片大小为480*360,GPU上的平均推理速度为:6ms

在这里插入图片描述

5.2 CPU推理

我测试的电脑为i7-12700H 2.3GHz,测试图片大小为480*360,CPU上的平均推理速度为:28ms

在这里插入图片描述

5.2.1 CPU推理代码修改

使用CPU推理需要修改的几个地方如下,修改好后再运行脚本test.py就是纯CPU推理了:

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

六、总结

以上就是AdaFM模型推理测试方法,我仅使用了官网提供的模型权重测试,没有训练,如学者需要训练,参考官网训练方法。有问题的欢迎留言讨论。

总结不易,多多支持,谢谢!


http://www.niftyadmin.cn/n/5263169.html

相关文章

jmeter如何循环运行到csv文件最后一行后停止

1、首先在线程组中设置’循环次数‘–勾选永远 2、csv数据文件设置中设置: 遇到文件结束符再次循环?——改为:False 遇到文件结束符停止线程?——改为:True 3、再次运行就会根据文档的行数运行数据 (如果需要在循环控制器中&…

量子技术将如何重构我们的生活

薛定谔的猫与量子世界的奥秘 在量子世界中,“薛定谔的猫”成为一个令人费解而神秘的概念,描述了生死叠加的状态。你能想通吗?想不通很正常,因为这是量子世界,是物理学最前沿的研究领域。在19世纪末,经典物理…

c语言多线程队列实现

为了用c语言实现队列进行多线程通信,用于实现一个状态机。 下面是实现过程 1.实现多线程队列入栈和出栈,不加锁 发送线程发送字符1,接收线程接收字符并打印。 多线程没有加锁,会有危险 #include "stdio.h" #include …

C++初阶(十六)优先级队列

📘北尘_:个人主页 🌎个人专栏:《Linux操作系统》《经典算法试题 》《C》 《数据结构与算法》 ☀️走在路上,不忘来时的初心 文章目录 一、priority_queue的介绍和使用1、priority_queue的介绍2、priority_queue的使用 二、priori…

2019年第八届数学建模国际赛小美赛D题安全选举的答案是什么解题全过程文档及程序

2019年第八届数学建模国际赛小美赛 D题 安全选举的答案是什么 原题再现: 随着美国进入一场关键性的选举,在确保投票系统的完整性方面进展甚微。2016年总统大选期间,唐纳德特朗普因被指控受到外国干涉而入主白宫,这一问题再次成为…

Flink 流处理流程 API详解

流处理API的衍变 Storm:TopologyBuilder构建图的工具,然后往图中添加节点,指定节点与节点之间的有向边是什么。构建完成后就可以将这个图提交到远程的集群或者本地的集群运行。 Flink:不同之处是面向数据本身的,会把D…

【排序算法】之归并排序

归并思想 先拆分后合并 也就是分治; 拆分合并思想具体讲解可以参考以下链接: b站链接: 点这里:b站归并思想具体讲解 看代码 代码中的例子参考上图和下图 public class MergeSort {//一、拆分部分public static void split(i…

【华为OD题库-092】单词加密-java

题目 输入一个英文句子,句子中包含若干个单词,每个单词间有一个空格需要将句子中的每个单词按照要求加密输出。要求: 1)单词中包括元音字符(‘aeuio’、‘AEUIO’,大小写都算),则将元音字符替换成’*) 2)单词中不包括元音字符&…