diff --git a/advanced/databases/mnist/5.py b/advanced/databases/mnist/5.py index a1bb0544a96e5fc3e799fcd21dd02ec6f3b12927..bb2dc0333b98259cd6aa07a0bc03e89116891260 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)) }