Zaktualizuj 'neural_network.py'

poprawka sieci neuronowych

neural_network.py

@@ -3,8 +3,8 @@ import cv2
 import numpy as np
 import matplotlib.pyplot as plt
 import tensorflow as tf
+import random
 
-def recognition():
 mnist = tf.keras.datasets.mnist
 (x_train, y_train), (x_test, y_test) = mnist.load_data()
 
@@ -24,8 +24,8 @@ def recognition():
 
 model = tf.keras.models.load_model('handwritten.model')
 
-    image_number = 1
-    while os.path.isfile(f"digits/digit{image_number}.png"):
+def recognition():
+    image_number = random.randint(1, 9)
     try:
         img = cv2.imread(f"digits/digit{image_number}.png")[:,:,0]
         img = np.invert(np.array([img]))
@@ -35,12 +35,9 @@ def recognition():
         plt.show()
     except:
         print("Error!")
-        finally:
-            image_number +=1
 
     loss, accuracy = model.evaluate(x_test, y_test)
 
     print(loss)
     print(accuracy)
-
 recognition()