Zaktualizuj 'neural_network.py'

This commit is contained in:
Aniela Walczak 2022-06-01 17:17:20 +02:00
parent 15234836ce
commit 1e24b39da3

View File

@ -29,7 +29,7 @@ def recognition():
digits=[]
try:
for i in range(0,3):
image_number = random.randint(1, 19)
image_number = random.randint(1, 17)
img = cv2.imread(f"digits/digit{image_number}.png")[:,:,0]
img = np.invert(np.array([img]))
prediction = model.predict(img)