Skip to content

GitLab

"tests/vscode:/vscode.git/clone" did not exist on "b117baf1b4dbf0602c36f6a5bc1816dc169312a7"
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
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
ptn/metrics.py
View file @ 614338f9
...@@ -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