对语义分割的输出进行阈值处理？

Question

我正在使用我用 UNet 架构构建的模型来分割显微图像。训练模型后我得到了合理的结果，所以我知道它是有效的。作为输出，我获得预测值在 0 到 1 之间的数组。一方面，这是有道理的，因为我使用了 sigmoid 激活函数（它应该产生 0 到 1 之间的输出）。然而，理想情况下每个值都是 0 或 1，就像我的目标图像中一样。我想知道是否应该在模型中的 sigmoid 激活之后添加一个阈值步骤，以使每个值都为 0 或 1，以便这是用于训练模型的值。我不确定这是否是人们所做的事情，我希望得到一些见解。

这是我得到的一个例子。事后我对输出进行了阈值处理：