Keras

Question

我正在使用

loss = 'mse'

在 Keras 中用于重建灰度图像的自动编码器模型。我的batch size是1。在训练过程中产生了一个单一的损失值。

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

如果输入多于一张灰度图像到模型中，这个过程是否相同？计算所有图像的平方像素差异，然后除以所有图像中的像素总数？

非常感谢

Answer 1

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

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

Keras - 图像的均方误差（MSE）计算定义？