计算图像的峰值信噪比并按表头写入EXCEL表格,代码
时间: 2024-06-11 15:04:01 浏览: 172
基于python与Django的网上购物平台
这里提供Python代码实现计算图像的峰值信噪比并将结果写入Excel表格。需要安装OpenCV和openpyxl库。
```
import cv2
import math
import openpyxl
# 读取图像
img = cv2.imread("test.png", cv2.IMREAD_GRAYSCALE)
# 计算峰值信噪比
max_pixel = 255
mse = (1.0 / (img.shape[0] * img.shape[1])) * sum(sum((img - cv2.mean(img)) ** 2))
psnr = 20 * math.log10(max_pixel / math.sqrt(mse))
# 写入Excel表格
wb = openpyxl.Workbook()
sheet = wb.active
sheet["A1"] = "图像名称"
sheet["B1"] = "峰值信噪比"
sheet["A2"] = "test.png"
sheet["B2"] = psnr
wb.save("result.xlsx")
```
其中,`test.png`为待计算峰值信噪比的图像名称,可以根据实际情况进行修改。运行代码后会在当前目录下生成一个名为`result.xlsx`的Excel表格,其中第一列为图像名称,第二列为对应的峰值信噪比。
阅读全文