From 359de79fa92e7c82320202262b5ed5171a421f25 Mon Sep 17 00:00:00 2001
From: Samuel Gaist <samuel.gaist@idiap.ch>
Date: Tue, 2 Feb 2021 11:11:38 +0100
Subject: [PATCH] [advanced][databases][mnist] Fix image loading
---
advanced/databases/mnist/5.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/advanced/databases/mnist/5.py b/advanced/databases/mnist/5.py
index a1bb054..bb2dc03 100644
--- a/advanced/databases/mnist/5.py
+++ b/advanced/databases/mnist/5.py
@@ -82,7 +82,7 @@ class View(BaseView):
elif output == 'image':
return {
- 'value': obj.image.reshape((28, 28))
+ 'value': np.array(obj.image).reshape((28, 28))
}
--
GitLab