在需要输出整个批次大小的 keras 中实现自定义成本函数

【问题标题】：Implement Custom Cost function in keras which requires output of whole batch-size在需要输出整个批次大小的 keras 中实现自定义成本函数
【发布时间】：2018-01-08 20:12:49
【问题描述】：

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

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

【问题讨论】：

标签： python tensorflow computer-vision keras

【解决方案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)

【讨论】：

感谢会尽力实施并回复您。