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=[] digits=[]
try: try:
for i in range(0,3): 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 = cv2.imread(f"digits/digit{image_number}.png")[:,:,0]
img = np.invert(np.array([img])) img = np.invert(np.array([img]))
prediction = model.predict(img) prediction = model.predict(img)