我正在尝试训练用于医学图像分割的slightly modified version of this UNet架构。
我现在对train.py文件中的88
行的功能有些困惑。 img_masks = np.array(img_masks > 0.45, dtype=int)
。
有人可以解释一下,如果遮罩是0
或1
,为什么要对遮罩进行阈值设置!这能实现什么?
我在训练包含这条线时的另一个观察结果是,我的dice coefficient scores
在每个时期都非常低(在1e-4
附近),并在阈值处进行了注释,我的骰子系数跃升到{{1}以上}据我所知,骰子得分应该在1
和0
之间。
我将不胜感激。