Skip to content

GitLab

Commit 4fb2e0cc authored by Oier Mees's avatar Oier Mees
Browse files

fixes voxel IOU evaluation

parent a380b4b3
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
ptn/metrics.py
View file @ 4fb2e0cc
...@@ -99,8 +99,8 @@ def add_volume_iou_metrics(inputs, outputs): ...@@ -99,8 +99,8 @@ def add_volume_iou_metrics(inputs, outputs):
names_to_updates = dict() names_to_updates = dict()
labels = tf.greater_equal(inputs['voxels'], 0.5) labels = tf.greater_equal(inputs['voxels'], 0.5)
predictions = tf.greater_equal(outputs['voxels_1'], 0.5) predictions = tf.greater_equal(outputs['voxels_1'], 0.5)
labels = 2 - tf.to_int32(labels) labels = (2 - tf.to_int32(labels)) - 1
predictions = 3 - tf.to_int32(predictions) * 2 predictions = (3 - tf.to_int32(predictions) * 2) - 1
tmp_values, tmp_updates = tf.metrics.mean_iou( tmp_values, tmp_updates = tf.metrics.mean_iou(
labels=labels, labels=labels,
predictions=predictions, 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