在keras中实现自定义成本函数，需要输出整批量

Question

有关数据集的简要信息： - 它是一个人的Re-ID问题（无论2张图像是否相同），我必须为自己的实现编写自己的损失/成本函数（二项式偏差）。 对于批量大小为128的i，将具有256个图像，因此具有3 256 * 256 /批量矩阵 我的疑问是如何使用keras获得批量大小的输出，这样我就可以实现成本函数

我添加了二项式偏差成本函数公式的截图。 enter image description here

Answer 1

使用额外的batch_size参数定义您的自定义损失：

def my_loss_template(y_true, y_pred, batch_size):
    # compute the loss

要使用它，

batch_size = 20
my_loss = lambda x, y: my_loss_template(x, y, batch_size)
model.compile(optimizer='adam', loss=my_loss)