在keras中实现Custom Cost功能，这需要输出整个批次的大小

Question

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

我添加了二项式偏差成本函数公式的屏幕截图。 在此处输入图片说明

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)