paddle官方提供的focal loss无法做到对背景类进行ignore,例如citysca里面的label255,在代码55行会报错,此时找不到索引。
解决方法:paddle.nn.functional.nll_loss(logpt, label, weight=None, ignore_index=self.ignore_index, reduction='mean', name=None)替换paddle.gather_nd(logpt, label)
ok
解决方法:paddle.nn.functional.nll_loss(logpt, label, weight=None, ignore_index=self.ignore_index, reduction='mean', name=None)替换paddle.gather_nd(logpt, label)
ok