Lab5.1 / metrics.py
AshWu03
final bugfixes and optimizations
a8b0b54
Raw History Blame Contribute Delete
560 Bytes
import numpy as np
from image_processor import preprocess_image
import config
def compute_mse(original, mosaic):
"""
Compute Mean Squared Error between original and mosaic.
Parameters:
-----------
original : np.ndarray
Original image
mosaic : np.ndarray
Mosaic image
Returns:
--------
float : MSE value
"""
# Preprocess image
original = preprocess_image(original, config.mosaic_size)
mse = np.mean((original.astype(float) - mosaic.astype(float)) ** 2)
return mse