Commit 614338f9 authored by Jasmine's avatar Jasmine Committed by GitHub
Browse files

Merge pull request #2282 from mees/fixEvalIOU#2239

 fixes Perspective Transformer IOU evaluation
parents 32105a38 4fb2e0cc
......@@ -99,8 +99,8 @@ def add_volume_iou_metrics(inputs, outputs):
names_to_updates = dict()
labels = tf.greater_equal(inputs['voxels'], 0.5)
predictions = tf.greater_equal(outputs['voxels_1'], 0.5)
labels = 2 - tf.to_int32(labels)
predictions = 3 - tf.to_int32(predictions) * 2
labels = (2 - tf.to_int32(labels)) - 1
predictions = (3 - tf.to_int32(predictions) * 2) - 1
tmp_values, tmp_updates = tf.metrics.mean_iou(
labels=labels,
predictions=predictions,
......
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