Commit 3e3a1da1 by Wenqi Li

Merge branch 'fix-multimodal-pipeline' into 'dev'

python2&3

See merge request !173
parents 0ff6498c 7c955739
Pipeline #10228 passed with stages
in 38 minutes 57 seconds
......@@ -435,7 +435,7 @@ def dice_dense(prediction, ground_truth, weight_map=None):
raise NotImplementedError
ground_truth = tf.cast(ground_truth, dtype=prediction.dtype)
# computing Dice over the spatial dimensions
reduce_axes = range(len(prediction.shape) - 1)
reduce_axes = list(range(len(prediction.shape) - 1))
dice_numerator = 2.0 * tf.reduce_sum(
prediction * ground_truth, axis=reduce_axes)
dice_denominator = \
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment