我编写了一个 python 脚本，它在逐像素分割期间计算每个类的中值频率平衡权重。然后，我Python Layer在caffe模型定义中添加了一个，它将权重发送到损失函数。基于此链接，用户提到SoftmaxWithLosscaffe 中的该层对应于 TensorFlow softmax_cross_entropy_with_logits。我的问题是如何将权重发送到SoftmaxWithLoss层？还可以使用哪些其他损失层median frequency balancing？我用过InfoGainLoss，但它不收敛。非常感谢您的帮助。
问问题
239 次
1 回答
0
如果您想"SoftmaxLoss"根据类别加权（即交叉熵损失），那么"InfogainLoss"这就是您所需要的。
请注意，几个月前在 caffe 中升级了 infogain 层：它现在集成"Softmax"到损失计算中以实现稳健的梯度估计。
于 2018-01-19T09:29:27.147 回答