首页 PaddleCV 帖子详情
focal loss使用问题 已解决
收藏
快速回复
PaddleCV 文章语义分割目标检测图像分类 638 2
focal loss使用问题 已解决
收藏
快速回复
PaddleCV 文章语义分割目标检测图像分类 638 2

paddle官方提供的focal loss无法做到对背景类进行ignore,例如citysca里面的label255,在代码55行会报错,此时找不到索引。

SK_qinghe
已解决
2# 回复于2021-09
解决方法:paddle.nn.functional.nll_loss(logpt, label, weight=None, ignore_index=self.ignore_index, reduction='mean', name=None)替换paddle.gather_nd(logpt, label)
展开
0
收藏
回复
全部评论(2)
时间顺序
SK_qinghe
#2 回复于2021-09

解决方法:paddle.nn.functional.nll_loss(logpt, label, weight=None, ignore_index=self.ignore_index, reduction='mean', name=None)替换paddle.gather_nd(logpt, label)

0
回复
余志良
#3 回复于2021-09

ok

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