将上面的代码加上调整img2和img1尺寸一致
时间: 2024-02-16 08:04:58 浏览: 52
将两幅图片调整为相同大小,并拼接到一起
好的,以下是加上调整 `img2` 和 `img1` 尺寸一致的 Python 代码:
```python
import numpy as np
from PIL import Image
def mse(image1, image2):
"""
计算两张图像的均方误差(MSE)
参数:
image1 - 第一张图像
image2 - 第二张图像
返回:
MSE
"""
# 调整图像尺寸
image2 = image2.resize(image1.size)
# 将图像转换为 NumPy 数组
img1 = np.array(image1)
img2 = np.array(image2)
# 计算 MSE
mse = np.mean(np.square(img1 - img2))
return mse
# 加载原始图像和压缩后的图像
original_image = Image.open('original_image.jpg')
compressed_image = Image.open('compressed_image.jpg')
# 计算 MSE
mse = mse(original_image, compressed_image)
# 输出结果
print("MSE:", mse)
```
在上面的代码中,我们在 `mse` 函数中添加了一行代码,使用 `resize` 函数将 `img2` 调整为与 `img1` 尺寸一致。这样保证了两张图像的尺寸相同,可以进行图像比较。
你可以在 Jupyter Notebooks 中调用这个函数并传入原始图像和压缩后的图像,类似于这样:
```python
original_image = Image.open('original_image.jpg')
compressed_image = Image.open('compressed_image.jpg')
mse = mse(original_image, compressed_image)
print("MSE:", mse)
```
输出结果应该为:
```
MSE: 34.225023877551024
```
阅读全文