X-Git-Url: https://piware.de/gitweb/?p=handwriting-recognition.git;a=blobdiff_plain;f=read_display_mnist.py;h=3030a0b50cc7b19142d6c1ca9b570e6c706e96b7;hp=fd85ee3887941fee01f3cd1259ea7b5a6dac55a5;hb=8af4223121b60d5d67b7121d87c5c6fed01b58e7;hpb=6bdea63ab81b6b3dae4649478d29563e2160bc13;ds=sidebyside

diff --git a/read_display_mnist.py b/read_display_mnist.py
index fd85ee3..3030a0b 100755
--- a/read_display_mnist.py
+++ b/read_display_mnist.py
@@ -10,5 +10,5 @@ train_images, train_labels, rows, cols = mnist.load('train-images-idx3-ubyte', '
 # show the first bunch of training data
 for i in range(10):
     print(f'train image #{i}: label {train_labels[i]}')
-    plt.imshow(train_images[i].reshape(rows, cols), cmap='gray')
+    plt.imshow(train_images[:, i].reshape(rows, cols), cmap='gray')
     plt.show()