python - 广义骰子损失问题

Question

我正在调整 niftynet 提供的广义骰子损失，以在我自己的 tensorflow 网络上使用，该网络产生单个标签的概率。为了使用广义骰子损失，我将这一行添加到文件的开头（我正在处理 2D 数据）：

prediction = tf.concat([1 - prediction, prediction], axis=3)

我使用的批量大小为 1。当我测试 Square、Simple 和 Uniform 的 type_weight 时，我得到了非常相似的结果。似乎应该对 ref_vol 求和，还是我错过了这个？

标签： pythontensorflowniftynet