search

bubbliiiing / hrnet-pytorch

这是一个hrnet-pytorch的库，可用于训练自己的语义分割数据集
MIT License
104 stars 16 forks source link

关于ignore_label #17

Open littlezbq opened 1 year ago

littlezbq commented 1 year ago

大佬好，我的数据集里面像素值为0的地方需要ignore掉，不参与loss的计算。我是不是在loss的相关函数里把ignore_index设置为0就可以了呢，我看您的代码那里默认的是num_class。

bubbliiiing commented 1 year ago

额0 0 也行……