标签: python tensorflow niftynet
我正在调整niftynet提供的广义骰子损失,以在我自己的张量流网络上使用,这会产生单个标签的概率。为了使用广义的骰子损失,我将以下行添加到文件的开头(我正在使用2D数据):
prediction = tf.concat([1 - prediction, prediction], axis=3)
我正在使用1的批处理大小。当我对Square,Simple和Uniform测试type_weight时,得到的结果非常相似。似乎应该对ref_vol求和,或者我错过了吗?