/tmp/ipykernel_3164/1425456350.py:11: RuntimeWarning: divide by zero encountered in double_scalars psnr = 10 * np.log10((255 ** 2) / mse)

这个警告通常是因为 MSE 的值为零，导致在计算 PSNR 时出现了除以零的情况。这可能是由于原始图像和压缩后的图像完全相同导致的。

为了避免这个警告，你可以在计算 PSNR 之前添加一个检查，判断 MSE 是否为零。如果 MSE 为零，说明原始图像和压缩后的图像完全相同，此时可以直接将 PSNR 设为一个较大的值。

下面是修改后的代码示例：

import cv2
import numpy as np

# 加载原始图像
img_original = cv2.imread('original_image.png', cv2.IMREAD_GRAYSCALE)

# 加载 JPEG2000 压缩后的图像
img_jp2 = cv2.imread('compressed_image.jp2', cv2.IMREAD_GRAYSCALE)

# 计算 MSE 值
mse = ((img_original - img_jp2) ** 2).mean()

# 计算 PSNR 值
if mse == 0:
    psnr = float('inf')
else:
    psnr = 10 * np.log10((255 ** 2) / mse)

print('PSNR:', psnr)

在上面的代码中，我们添加了一个 if-else 语句来检查 MSE 是否为零，并根据情况计算 PSNR 值。如果 MSE 为零，说明原始图像和压缩后的图像完全相同，此时将 PSNR 设为正无穷大。