广义骰子损失问题

时间:2018-09-03 09:49:03

标签: python tensorflow niftynet

我正在调整niftynet提供的广义骰子损失,以在我自己的张量流网络上使用,这会产生单个标签的概率。为了使用广义的骰子损失,我将以下行添加到文件的开头(我正在使用2D数据):

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

我正在使用1的批处理大小。当我对Square,Simple和Uniform测试type_weight时,得到的结果非常相似。似乎应该对ref_vol求和,或者我错过了吗?

0 个答案:

没有答案