diff --git a/vis/view_results.html b/vis/view_results.html
index 589c469..12d59b2 100644
--- a/vis/view_results.html
+++ b/vis/view_results.html
@@ -15,6 +15,7 @@
var INPUT_JSON_PATH = 'data/results.json'
var input_struct = null; // will be loaded from json
var current_id = 0; // currently shown image id
+ var show_scores = 0;
// render flags utils
var render_flags = {
@@ -29,6 +30,13 @@
render_flags[name] = !render_flags[name];
renderAnnotations();
}
+ function toggleScores() {
+ show_scores = !show_scores;
+ renderAnnotations();
+ }
+ function sigmoid(t) {
+ return 1/(1+Math.pow(Math.E, -t));
+ }
function getUrlParam(name, fallback) {
name = name.replace(/[\[]/, "\\[").replace(/[\]]/, "\\]");
@@ -111,11 +119,16 @@
var i = ixscore[k][0];
var det_box = result.boxes[i];
var caption = result.captions[i];
+ var score = sigmoid(result.scores[i]);
// render box
var color = WAD_COLORS[k % WAD_COLORS.length];
renderBox(elt, det_box, color, BOX_WIDTH, render_flags.captions_inline ? caption : '');
// render caption in a separate div below the image
- delt.append('span').classed('ddesc', true).style('color', color).html(escapeHtml(caption) + '. ');
+ if (show_scores) {
+ delt.append('span').classed('ddesc', true).style('color', color).html(escapeHtml(caption) + ' (' + score.toFixed(2) + '). ');
+ } else {
+ delt.append('span').classed('ddesc', true).style('color', color).html(escapeHtml(caption) + '. ');
+ }
}
}
@@ -127,6 +140,7 @@
if (e.keyCode == 83) updateCounter('detections_to_show', -1);
if (e.keyCode == 87) updateCounter('detections_to_show', 1);
if (e.keyCode == 84) toggleFlag('captions_inline');
+ if (e.keyCode == 67) toggleScores();
if (e.keyCode == 82) jumpRandom();
};
@@ -159,6 +173,9 @@
+