在训练完全卷积网络时忽略标签

作者: 氷寒
发布时间: 2024-12-23 01:37:21 (1月前)
转自：

2 条回复

0#
回复此人
12345 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 使用，例如 <a href="http://caffe.help/manual/layers/softmaxwithloss.html" rel="nofollow noreferrer"> <code> "SoftmaxWithLoss" </code> </A> 图层，你可以添加一个 <code> loss_param { ignore_label: 255 } </code> 告诉caffe忽略这个标签： </p> <pre> <code> layer { name: "loss" type: "SoftmaxWithLoss" bottom: "prediction" bottom: "labels_with_255_as_ignore" loss_weight: 1 loss_param: { ignore_label: 255 } } </code> </pre> <P> 我没有检查，但我相信 <code> ignore_label </code> 也用于 <code> InfogainLoss </code> 损失和一些其他损失层。 </p> </DIV>

编辑

登录后才能参与评论