test image reshaping

danielhallinan88 · Apr 15, 2020 · 4ea5d91 · 4ea5d91
1 parent 5b7f6d6
commit 4ea5d91
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/app.py b/app.py
@@ -97,6 +97,11 @@ def mnist():
 
             img        = Image.open(data)
             img_arr    = np.array(img, dtype='float32')
+            #with open('shapes.log','w') as fh:
+            #    fh.write("Shape: {}\n".format(img_arr.shape))
+            #    rgb_weights = [0.2989, 0.5870, 0.1140]
+            #    grayscale_image = np.dot(img_arr[...,:3], rgb_weights)
+            #    fh.write("Gray Shape: {}".format(grayscale_image.shape))
             img_arr_sh = img_arr.reshape(1, 28, 28, 1)
             prediction = model.predict(img_arr_sh)
             result     = np.argmax(prediction)