Keras - 图像的均方误差 (MSE) 计算定义？

Question

我正在使用

loss = 'mse' 

在 Keras 中用于重建灰度图像的自动编码器模型。我的批量大小是 1。在训练期间产生了一个损失值。

我似乎无法在任何地方找到这个问题的答案。 Keras 如何计算这两张图片的 MSE 损失值？它们表示为 2d NumPy 数组。它是否计算每个像素之间的平方差然后除以像素数（考虑批量大小为1）？

如果输入超过1张灰度图像进入模型，这个过程是否相同；计算所有图像的像素差平方，然后除以所有图像的像素总数？

非常感谢

Answer 1

def mse(y_true, y_pred):
    return K.mean(K.square(y_pred - y_true), axis=-1)

这是 mse 的代码，操作（差分和平方）是按位（逐个像素）的，然后计算平均值，然后除以值（像素）的数量。