什么损耗层可用于中位频率平衡？怎么样？

Question

我编写了一个python脚本，用于计算逐像素分割期间每个类的中值频率平衡权重。然后，我在Python Layer模型定义中添加了一个caffe，它将权重发送到损失函数。基于this link，用户提到caffe中的SoftmaxWithLoss层对应于TensorFlow softmax_cross_entropy_with_logits。我的问题是如何将权重发送到SoftmaxWithLoss层？ Qazxswpoi可以使用哪些其他损耗图层？我使用median frequency balancing，但它没有收敛。非常感谢您的帮助。

Answer 1

如果你想根据类别加权InfoGainLoss（即交叉熵损失），那么"SoftmaxLoss"就是你需要的。 请注意，几个月前infogain层在caffe中进行了升级：它现在将"InfogainLoss"集成到损失计算中，以实现稳健的梯度估计。