From 730c4a7c28ab5937e675a3f7e3e238544da20b82 Mon Sep 17 00:00:00 2001 From: Michele Toni Date: Mon, 27 Mar 2017 14:02:13 +0200 Subject: [PATCH 1/2] FCN confusion matrix visualization fix Fix LINT check --- digits/model/images/classification/views.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/digits/model/images/classification/views.py b/digits/model/images/classification/views.py index 9d5d12012..3a8f0a4d1 100644 --- a/digits/model/images/classification/views.py +++ b/digits/model/images/classification/views.py @@ -529,6 +529,11 @@ def classify_many(): 'Unable to classify any image from the file') scores = last_output_data + # force correct 2D shape squeezing scores + for i in reversed(range(2, len(scores.shape))): + if scores.shape[i] == 1: + scores = np.squeeze(scores, axis=(i,)) + # take top 5 indices = (-scores).argsort()[:, :5] From 058e267d8fc3336049770965f4c215cee3fa4c4d Mon Sep 17 00:00:00 2001 From: Alessandro Beltramin Date: Thu, 30 Mar 2017 20:14:32 +0200 Subject: [PATCH 2/2] FCN TOPN visualization fix --- digits/model/images/classification/views.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/digits/model/images/classification/views.py b/digits/model/images/classification/views.py index 3a8f0a4d1..d843d8912 100644 --- a/digits/model/images/classification/views.py +++ b/digits/model/images/classification/views.py @@ -670,6 +670,11 @@ def top_n(): if scores is None: raise RuntimeError('An error occurred while processing the images') + # force correct 2D shape squeezing scores + for i in reversed(range(2, len(scores.shape))): + if scores.shape[i] == 1: + scores = np.squeeze(scores, axis=(i,)) + labels = model_job.train_task().get_labels() images = inputs['data'] indices = (-scores).argsort(axis=0)[:top_n]