python计算psnr和ssim

时间: 2023-08-03 11:08:05 浏览: 117
你可以使用Python中的OpenCV库来计算图像的PSNR(峰值信噪比)和SSIM(结构相似性指数)。 首先,确保你已经安装了OpenCV库。可以使用以下命令安装: ``` pip install opencv-python ``` 接下来,你可以使用以下代码计算PSNR和SSIM: ```python import cv2 import numpy as np def calculate_psnr(original_image, compressed_image): mse = np.mean((original_image - compressed_image) ** 2) if mse == 0: return float('inf') max_pixel = 255.0 psnr = 20 * np.log10(max_pixel / np.sqrt(mse)) return psnr def calculate_ssim(original_image, compressed_image): ssim = cv2.SIFT_score(original_image, compressed_image) return ssim # 读取原始图像和压缩后的图像 original_image = cv2.imread('original_image.jpg') compressed_image = cv2.imread('compressed_image.jpg') # 将图像转为灰度图像 original_gray = cv2.cvtColor(original_image, cv2.COLOR_BGR2GRAY) compressed_gray = cv2.cvtColor(compressed_image, cv2.COLOR_BGR2GRAY) # 计算PSNR和SSIM psnr = calculate_psnr(original_gray, compressed_gray) ssim = calculate_ssim(original_gray, compressed_gray) print(f"PSNR: {psnr} dB") print(f"SSIM: {ssim}") ``` 请确保将 `original_image.jpg` 和 `compressed_image.jpg` 替换为你实际的图像文件路径。这段代码将计算灰度图像的PSNR和SSIM,并输出结果。 希望这可以帮到你!如果有任何疑问,请随时问我。

相关推荐

最新推荐

recommend-type

6种图像评价原理(UCIQE、UICM、PSNR、SSIM、等效圆偏检测、图像信息熵)

6种图像评价原理(UCIQE、UICM、PSNR、SSIM、等效圆偏检测、图像信息熵)
recommend-type

浔川AI翻译技术建设社团

https://blog.csdn.net/2401_83104529/article/details/139215262?spm=1001.2014.3001.5501
recommend-type

llama-factory一个数据微调用例

llama-factory一个数据微调用例
recommend-type

YOLOV5 实战项目:辣椒缺陷检测(2类别)【数据+代码+训练好的权重】

基于YOLOV5 对辣椒缺陷检测(2类别)的目标检测实战项目,包含代码、数据集、训练好的权重参数,经测试,代码可以直接使用 图像分辨率为640*640的大分辨率RGB图片,数据集为采摘好的辣椒好坏检测。标注的边界框完整,每张图像均有多个目标。 【数据】(分为分为训练集和验证集) 训练集datasets-images-train:364张图片和364个标签txt文件组成 验证集datasets-images-val:88张图片和88个标签txt文件组成 【yolov5】项目总大小:62MB 项目检测测试了100个epoch,在runs目录下保存了训练结果,训练最好的精度map0.5=0.92,map0.5:0.95=0.89。网络还没收敛,加大epoch可以得到更好的结果。 训练过程中会生成验证集的混淆矩阵,PR曲线、F1曲线等等 更多yolov5改进介绍、或者如何训练,请参考: https://blog.csdn.net/qq_44886601/category_12605353.html
recommend-type

zigbee-cluster-library-specification

最新的zigbee-cluster-library-specification说明文档。
recommend-type

管理建模和仿真的文件

管理Boualem Benatallah引用此版本:布阿利姆·贝纳塔拉。管理建模和仿真。约瑟夫-傅立叶大学-格勒诺布尔第一大学,1996年。法语。NNT:电话:00345357HAL ID:电话:00345357https://theses.hal.science/tel-003453572008年12月9日提交HAL是一个多学科的开放存取档案馆,用于存放和传播科学研究论文,无论它们是否被公开。论文可以来自法国或国外的教学和研究机构,也可以来自公共或私人研究中心。L’archive ouverte pluridisciplinaire
recommend-type

MATLAB柱状图在信号处理中的应用:可视化信号特征和频谱分析

![matlab画柱状图](https://img-blog.csdnimg.cn/3f32348f1c9c4481a6f5931993732f97.png) # 1. MATLAB柱状图概述** MATLAB柱状图是一种图形化工具,用于可视化数据中不同类别或组的分布情况。它通过绘制垂直条形来表示每个类别或组中的数据值。柱状图在信号处理中广泛用于可视化信号特征和进行频谱分析。 柱状图的优点在于其简单易懂,能够直观地展示数据分布。在信号处理中,柱状图可以帮助工程师识别信号中的模式、趋势和异常情况,从而为信号分析和处理提供有价值的见解。 # 2. 柱状图在信号处理中的应用 柱状图在信号处理
recommend-type

前端深拷贝 和浅拷贝有哪些方式,你在哪里使用过

前端深拷贝和浅拷贝的方式有很多,下面列举几种常用的方式: 深拷贝: 1. JSON.parse(JSON.stringify(obj)),该方法可以将对象序列化为字符串,再将字符串反序列化为新的对象,从而实现深拷贝。但是该方法有一些限制,例如无法拷贝函数、RegExp等类型的数据。 2. 递归拷贝,即遍历对象的每个属性并进行拷贝,如果属性值是对象,则递归进行拷贝。 3. 使用第三方库如lodash、jQuery等提供的深拷贝方法。 浅拷贝: 1. Object.assign(target, obj1, obj2, ...),该方法可以将源对象的属性浅拷贝到目标对象中,如果有相同的属性,则会
recommend-type

JSBSim Reference Manual

JSBSim参考手册,其中包含JSBSim简介,JSBSim配置文件xml的编写语法,编程手册以及一些应用实例等。其中有部分内容还没有写完,估计有生之年很难看到完整版了,但是内容还是很有参考价值的。
recommend-type

"互动学习:行动中的多样性与论文攻读经历"

多样性她- 事实上SCI NCES你的时间表ECOLEDO C Tora SC和NCESPOUR l’Ingén学习互动,互动学习以行动为中心的强化学习学会互动,互动学习,以行动为中心的强化学习计算机科学博士论文于2021年9月28日在Villeneuve d'Asq公开支持马修·瑟林评审团主席法布里斯·勒菲弗尔阿维尼翁大学教授论文指导奥利维尔·皮耶昆谷歌研究教授:智囊团论文联合主任菲利普·普雷教授,大学。里尔/CRISTAL/因里亚报告员奥利维耶·西格德索邦大学报告员卢多维奇·德诺耶教授,Facebook /索邦大学审查员越南圣迈IMT Atlantic高级讲师邀请弗洛里安·斯特鲁布博士,Deepmind对于那些及时看到自己错误的人...3谢谢你首先,我要感谢我的两位博士生导师Olivier和Philippe。奥利维尔,"站在巨人的肩膀上"这句话对你来说完全有意义了。从科学上讲,你知道在这篇论文的(许多)错误中,你是我可以依