Zaktualizuj 'neural_network.py'

This commit is contained in:
Aniela Walczak 2022-06-01 17:21:46 +02:00
parent 88dd3cf7ab
commit 1f17a731b2

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, 17) image_number = random.randint(1, 19)
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)