关于语义分割损失函数忽略label的问题
收藏
使用Cityscapes数据集,label像素变成了0-19共20类,所以 num_class = 20,
但是像素值为19的是我不想关注的类别,我只想关注像素值0-18的类,
然后损失函数使用的是 softmax + cross_entropy是分开写的,我用的softmax的api,在用的cross_entropy的api,
发现 paddle.fluid.layers.cross_entropy(input, label, soft_label=False, ignore_index=-100) API中有 ignore_index忽略此标签值,不参与运算。
那么我将 ignore_index = 19 是不是就可以认为 我在计算中忽略了 像素值为19的类
0
收藏
请登录后评论
这个API应该是忽略输入图片的像素值,而不是图片中的像素
cityscapes实际预测类别数应该是19类(0-18),ignore_index=255(255是你在训练前及那个label做了减一处理)