unet-pytorch icon indicating copy to clipboard operation
unet-pytorch copied to clipboard

loss计算

Open yinshuisiyuan123 opened this issue 3 years ago • 9 comments

导师,我的数据集缺陷数据集,就两个类别,前景(缺陷),背景,label的像素点是0和1,在数据增强的时候对label用0填充 image 数据增强以后,我的label的像素点还是0和1啊,那下面代码的意义何在? image 如此,在计算损失的时候,我的num_class=2,ignore_index= 2就没有任何作用啊,计算的损失还是包含了我数据增强填充的0 image

很想请教这个损失问题,谢谢UP

yinshuisiyuan123 avatar Jul 03 '22 11:07 yinshuisiyuan123

可能是我水平有限,二分类问题在计算时不需要考虑???

yinshuisiyuan123 avatar Jul 03 '22 11:07 yinshuisiyuan123

不是只针对缺陷数据集设计的代码 你应该看看voc数据集

bubbliiiing avatar Jul 03 '22 14:07 bubbliiiing

不是只针对缺陷数据集设计的代码 你应该看看voc数据集 针对自己的数据集,这部分是否需要修改呢?

yinshuisiyuan123 avatar Jul 03 '22 14:07 yinshuisiyuan123

这并不影响你的数据集

bubbliiiing avatar Jul 03 '22 14:07 bubbliiiing

导师,我只是疑惑,如果按照您的代码,这个loss怎么算啊?在计算的loss的时候,label岂不是包含我数据增强部分的填充?希望导师指教一下,感谢!

yinshuisiyuan123 avatar Jul 03 '22 14:07 yinshuisiyuan123

1、按照我的代码,loss就是图中这样算 2、”label岂不是包含我数据增强部分的填充?“正常很难理解你是什么意思

bubbliiiing avatar Jul 03 '22 14:07 bubbliiiing

谢谢导师,我再debugg一下

yinshuisiyuan123 avatar Jul 03 '22 14:07 yinshuisiyuan123

嗯,希望你理解一下,这段大于的代码在你的场景里没有意义,更没有必要纠结

bubbliiiing avatar Jul 03 '22 14:07 bubbliiiing

嗯,谢谢导师

yinshuisiyuan123 avatar Jul 03 '22 14:07 yinshuisiyuan123