发布时间:2026/8/23 16:53:13
Invertible-Image-Rescaling 评测指南:PSNR/SSIM 脚本使用与 2x/4x/8x 基准数据集结果复现 Invertible-Image-Rescaling 评测指南PSNR/SSIM 脚本使用与 2x/4x/8x 基准数据集结果复现【免费下载链接】Invertible-Image-Rescaling[ECCV 2020, IJCV 2022] Invertible Image Rescaling项目地址: https://gitcode.com/gh_mirrors/in/Invertible-Image-RescalingInvertible-Image-RescalingIRN是 ECCV 2020 Oral 论文《Invertible Image Rescaling》的官方 PyTorch 实现用单一可逆网络同时完成图像的下采样与上采样效果远超传统下采样 超分辨率的组合方案。本文是一份面向新手的完整评测指南手把手教你使用项目自带的 PSNR/SSIM 脚本并复现在 Set5、Set14、BSD100、Urban100、DIV2K 等基准数据集上的 2x/4x/8x 重建指标。一、先认识 IRN可逆图像重缩放网络IRN 的核心思想很优雅一个 InvBlock 可逆块先通过 Haar 变换提取细节再逐级缩小/放大特征图最后由 Downscaling Module 输出低分辨率图像。因为网络是可逆的同一套权重既能下采样也能上采样。 复现评测时只需关注测试环节训练代码同样内置在仓库中如 codes/train.py但本文聚焦评测与结果复现。二、环境准备一键装好依赖推荐在 Anaconda 中创建 Python 3 虚拟环境pip install torch torchvision numpy opencv-python lmdb pyyaml依赖说明见 README.md 的Dependencies and Installation一节。有 NVIDIA GPU 的机器体验最佳配置文件中默认gpu_ids: [0]。三、准备基准数据集IRN 测试常用的 5 个基准数据集数据集规模特点Set55 张最常用、最快出结果Set1414 张含更多纹理细节BSD100100 张标准基准Urban100100 张城市建筑边缘密集DIV2K (val)100 张大分辨率验证集数据集下载方式可参考 BasicSR 的 Dataset Preparation 文档见 README.md 的Dataset Preparation。下载后按数据集名建好文件夹并把路径填进测试配置即可。四、PSNR/SSIM 独立评测脚本使用方法如果你已经有了生成图像例如自己跑出的结果目录可以直接用仓库提供的独立脚本 metrics/calculate_PSNR_SSIM.py 计算指标其结果与 MATLAB 官方算法完全一致。4.1 修改脚本头部配置脚本采用改配置直接运行的设计打开文件只需改 4 个变量folder_GT /path/to/Set5 # GT原图文件夹 folder_Gen /path/to/results # 生成/恢复图像文件夹 crop_border 4 # 评测前裁剪的边框像素 test_Y False # True: 只评 Y 通道; False: 评 RGB4.2 运行并解读输出python metrics/calculate_PSNR_SSIM.py运行后逐张打印每幅图的指标最后一行输出平均值Average: PSNR: 43.991234 dB, SSIM: 0.987123两个关键细节crop_border 4这是社区惯例裁掉 4 个边缘像素以排除边界效应论文结果均按此口径统计test_Y True切换到 YCbCr 的 Y 通道评测论文 PSNR_Y 指标即此口径RGB 与 Y 通道结果略有差异对比论文数据时请注意口径一致。脚本内部的核心函数calculate_psnr基于 MSE 计算calculate_ssim使用 11×11 高斯窗口实现细节可参考 calculate_PSNR_SSIM.py。五、一键复现 2x/4x/8x 测试test.py 配置文件IRN 更推荐的方式是直接运行测试脚本 codes/test.py它会边推理边自动计算并打印 PSNR/SSIM无需额外跑独立脚本。5.1 配置测试任务测试配置位于 codes/options/test/ 目录覆盖了全部常用倍数配置文件缩放倍数说明test_IRN_x2.yml2x快速验证首选test_IRN_x3.yml3x中间倍数test_IRN_x4.yml4x论文主力设置test_IRN_x8.yml8x大倍数下采样打开对应 yml把 5 个dataroot_GT填成本地数据集路径即可例如datasets: test_1: name: set5 mode: LQGT dataroot_GT: /path/to/Set5 # 只改这一行同时确认path.pretrain_model_G指向已下载好的权重文件如IRN_x4.pth预训练模型可在代码框架说明 codes/README.md 中查看获取方式。5.2 启动测试在codes/目录下执行python test.py -opt options/test/test_IRN_x4.yml5.3 看懂输出日志每处理一张图日志会打印两类指标101 - PSNR: 45.210344 dB; SSIM: 0.991234; PSNR_Y: 45.778123 dB; SSIM_Y: 0.991834. ----Average PSNR/SSIM results for set5---- psnr: 43.991234 db; ssim: 0.987123.PSNR / SSIMRGB 全通道口径PSNR_Y / SSIM_YY 通道口径论文主表采用日志末尾的Average行就是你要汇报的基准成绩结果保存在results/目录中。⚠️ 复现提示IRN 下采样含随机采样 z论文报告的是 5 次采样的均值单次运行与论文数值可能有 0.02 dB 的微小波动属正常现象。六、基准数据集结果参考值IRN vs 其他方法下表为论文在 2x / 4x 下的核心结果PSNR / SSIM可作为你复现结果的对照基准2x 下采样 上采样方法Set5Set14BSD100Urban100DIV2KBicubic Bicubic33.66 / 0.929930.24 / 0.868829.56 / 0.843126.88 / 0.840331.01 / 0.9393Bicubic EDSR38.20 / 0.960634.02 / 0.920432.37 / 0.901833.10 / 0.936335.12 / 0.9699IRN (ours)43.99 / 0.987140.79 / 0.977841.32 / 0.987639.92 / 0.986544.32 / 0.99084x 下采样 上采样方法Set5Set14BSD100Urban100DIV2KBicubic Bicubic28.42 / 0.810426.00 / 0.702725.96 / 0.667523.14 / 0.657726.66 / 0.8521Bicubic ESRGAN32.74 / 0.901229.00 / 0.791527.84 / 0.745527.03 / 0.815230.92 / 0.8486IRN (ours)36.19 / 0.945132.67 / 0.901531.64 / 0.882631.41 / 0.915735.07 / 0.9318可以看出 IRN 仅用 1.66M2x/ 4.35M4x参数就全面超越动辄 15M 参数的专用下采样 专用超分组合在 Set5 4x 上甚至把 PSNR 拉高到 36.19 dB优势非常明显。完整对比表可参考 README.md 与上文插图。七、常见问题与复现清单结果和论文差一点确认三件事①crop_border取 4或等于 scale② 对比 PSNR 时明确是 RGB 还是 Y 通道③ 多次运行取平均。只想快速验证流程先用最小的 Set5 test_IRN_x2.yml几分钟即可跑通。生成图像存哪里results_root下按数据集名分文件夹命名如001.png方便后续用 calculate_PSNR_SSIM.py 复核。想复现颜色化或压缩扩展实验分别使用 test_IRN-Color.py 与 test_IRN-Compression.py 及对应 options/test/ 配置。八、总结本文带你走完了 Invertible-Image-Rescaling 的完整评测链路用独立脚本 metrics/calculate_PSNR_SSIM.py 对任意结果目录算 PSNR/SSIM用 codes/test.py codes/options/test/ 中现成的 2x/4x/8x 配置文件一键复现基准成绩对照论文表格核对 Set5 / Set14 / BSD100 / Urban100 / DIV2K 的指标。跟随以上步骤你应当能得到与论文一致±0.02 dB 内的 PSNR/SSIM 结果完成对可逆图像重缩放模型的标准评测闭环。 【免费下载链接】Invertible-Image-Rescaling[ECCV 2020, IJCV 2022] Invertible Image Rescaling项目地址: https://gitcode.com/gh_mirrors/in/Invertible-Image-Rescaling创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

相关新闻

2026/8/23 18:18:19

在不同空间中展示数据(欧式、球面、庞加莱圆盘)

文章目录欧式空间超球面双曲空间本文参照了Geomstats的官方教程,对不同流形上的数据进行了展示。 欧式空间 我们平时接触最多的就是欧式空间,在geomstats中,可创建一个Euclidean对其实现。下面创建一个欧氏空间,并计算其上两个随…

2026/8/23 18:18:19

SAP ME核心价值解析:从MES选型到与S/4HANA深度集成实战

如果你在制造业做信息化,特别是负责生产车间这一层,那么“MES”这个词你一定不陌生。但当你真正去选型、实施或开发时,往往会陷入困惑:市面上的MES五花八门,有的像ERP的延伸,有的像数据采集器的集合&#x…

2026/8/23 18:18:19

AI智能体与设计语法:重塑实验科学范式的核心架构

1. 从“炼丹”到“设计”:AI智能体如何重塑实验科学范式如果你在实验室里泡过,或者做过任何需要反复试错、调整参数的实验工作,你肯定对“炼丹”这个词不陌生。它精准地描述了传统实验科学中那种高度依赖直觉、经验和运气的探索过程——我们设…

2026/8/23 18:18:19

大厂面试必考的Java核心技术解析与实战技巧

1. 为什么大厂面试总爱考八股文? 刚入行那会儿,我也特别不理解为什么大厂面试总喜欢问那些看似死记硬背的"八股文"问题。直到后来自己当了面试官才明白,这些看似基础的问题,实际上是在考察候选人对Java技术体系的理解深…

2026/8/23 18:18:19

VisuScratch抓挠行为分析系统:从计算机视觉原理到科研实践全指南

在神经科学、药理学和疼痛行为学研究中,准确量化实验动物的抓挠行为是评估瘙痒、疼痛模型以及药物疗效的关键环节。传统的人工观察和计数方法不仅耗时耗力,主观性强,且难以捕捉行为的细微动态。VisuScratch 大小鼠抓挠行为分析系统正是为了解…

2026/8/23 18:13:19

FPGA 面试常问 FIFO:同步 FIFO、异步 FIFO、almost full 怎么答

前言FIFO 是 FPGA 面试里非常高频的问题。本文从新人视角讲清楚 FIFO 的基本作用、同步 FIFO 和异步 FIFO 的区别、空满判断方法、异步 FIFO 为什么要用 Gray 码,以及 almost full 到底是干什么的。最后给出一套面试时可以直接套用的回答模板,避免只会背…

2026/8/23 0:02:04

[光学原理与应用-521]:对光的错误理解与纠偏

首先光是一种能量的载体和形态,宏观上观察到的光是由无数个微观的光量子组成的,每个光子在产生的瞬间,其在真空的空间中以确定不变的速度沿着一个初始的方向一直向前,在微观层面,每个光量子的运动轨迹是以波函数所展现…

2026/8/23 0:02:04

SIP通话转接原理与REFER方法实战解析

1. 通话转接不是“挂断再拨号”,而是SIP会话的动态重定向你有没有遇到过这样的场景:客服坐席A正在和客户通电话,突然需要把这通对话无缝转给专家坐席B,客户完全感知不到中间的断连——既没听到忙音,也没被要求重新拨号…

2026/8/23 0:02:04

Kolla-ansible单节点OpenStack部署实战:从环境准备到排坑指南

1. 为什么选择Kolla-ansible来部署单节点OpenStack?如果你正在寻找一种能把OpenStack从“概念”快速变成“可用的实验环境”的方法,那么Kolla-ansible几乎是当前最主流、最省心的选择。我见过太多人卡在手动编译依赖、配置服务、处理版本冲突的泥潭里&am…

2026/8/23 0:02:04

[光学原理与应用-521]:对光的错误理解与纠偏

首先光是一种能量的载体和形态,宏观上观察到的光是由无数个微观的光量子组成的,每个光子在产生的瞬间,其在真空的空间中以确定不变的速度沿着一个初始的方向一直向前,在微观层面,每个光量子的运动轨迹是以波函数所展现…

2026/8/23 0:02:04

SIP通话转接原理与REFER方法实战解析

1. 通话转接不是“挂断再拨号”,而是SIP会话的动态重定向你有没有遇到过这样的场景:客服坐席A正在和客户通电话,突然需要把这通对话无缝转给专家坐席B,客户完全感知不到中间的断连——既没听到忙音,也没被要求重新拨号…

2026/8/23 0:02:04

Kolla-ansible单节点OpenStack部署实战:从环境准备到排坑指南

1. 为什么选择Kolla-ansible来部署单节点OpenStack?如果你正在寻找一种能把OpenStack从“概念”快速变成“可用的实验环境”的方法,那么Kolla-ansible几乎是当前最主流、最省心的选择。我见过太多人卡在手动编译依赖、配置服务、处理版本冲突的泥潭里&am…

2026/8/23 13:29:45

实测才敢推 AI论文网站 2026最新测评与推荐

2026年真正好用的AI论文网站,核心看生成的论文质量、低AI味、格式正确、学术适配四大指标。综合实测,千笔AI、ThouPen、豆包、DeepSeek、Grammarly 是当前最值得推荐的梯队,覆盖从免费到付费、从中文到英文、从文科到理工的全场景需求。一、综…

2026/8/23 6:14:43

2026必备!AI论文网站测评:最新推荐与深度对比

2026年真正好用的AI论文网站,核心看生成的论文质量、低AI味、格式正确、学术适配四大指标。综合实测,千笔AI、ThouPen、豆包、DeepSeek、Grammarly 是当前最值得推荐的梯队,覆盖从免费到付费、从中文到英文、从文科到理工的全场景需求。 一、…

2026/8/23 4:22:01

摆脱论文困扰!盘点2026年全网爆红的的AI论文写作工具

一天写完毕业论文在2026年已不再是天方夜谭。2026年最炸裂、实测能大幅提速的AI论文写作工具,覆盖选题构思、文献整理、内容生成、格式排版等核心场景,真正帮你高效搞定论文难题。 一、全流程王者:一站式搞定论文全链路(一天定稿首…