We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 32105a3 + 4fb2e0c commit 614338fCopy full SHA for 614338f
ptn/metrics.py
@@ -99,8 +99,8 @@ def add_volume_iou_metrics(inputs, outputs):
99
names_to_updates = dict()
100
labels = tf.greater_equal(inputs['voxels'], 0.5)
101
predictions = tf.greater_equal(outputs['voxels_1'], 0.5)
102
- labels = 2 - tf.to_int32(labels)
103
- predictions = 3 - tf.to_int32(predictions) * 2
+ labels = (2 - tf.to_int32(labels)) - 1
+ predictions = (3 - tf.to_int32(predictions) * 2) - 1
104
tmp_values, tmp_updates = tf.metrics.mean_iou(
105
labels=labels,
106
predictions=predictions,
0 commit comments