unet-pytorch
unet-pytorch copied to clipboard
dice loss 的一个小疑问
博主您好,非常感谢您的开源代码,我在数据读取的地方有一个小疑问,就是dice loss对应的标签seg_labels的问题。seg_labels对应的是one-shot形式,其中num_classes已经是加一的了,为什么该部分还要再加一呢? 如能回答,万分感谢呀,这地方没弄懂。
哪里加1了。。我看了半天没看出来