我正在尝试训练这个 UNet 架构的略微修改版本,用于医学图像分割。
我现在88
对 train.py 文件中 line 的功能有点困惑。 img_masks = np.array(img_masks > 0.45, dtype=int)
.
有人可以解释为什么如果口罩是0
或1
!这有什么作用?
我在使用该行进行训练时的另一个观察结果是,我dice coefficient scores
在每个时期(Around)都非常低,1e-4
并且将阈值注释掉,我的骰子系数跳到了上面1
,就我的理解而言,骰子得分应该在0
和之间1
。
我会很感激任何帮助。