首页 PaddleCV 帖子详情
关于语义分割损失函数忽略label的问题
收藏
快速回复
PaddleCV 问答语义分割 3176 2
关于语义分割损失函数忽略label的问题
收藏
快速回复
PaddleCV 问答语义分割 3176 2

使用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
收藏
回复
全部评论(2)
时间顺序
什么都不懂的人
#2 回复于2020-12

这个API应该是忽略输入图片的像素值,而不是图片中的像素

0
回复
SK_qinghe
#3 回复于2021-12

cityscapes实际预测类别数应该是19类(0-18),ignore_index=255(255是你在训练前及那个label做了减一处理)

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户