s464943/SI_InteligentnyWozekWidlowy
4
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

small fix

Browse Source
This commit is contained in:
Jakub-Prus 2022-06-06 23:46:27 +02:00
parent e5a7a975e8
commit 566a8cd868
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
imageClasification/Classificator.py
View File

@ -16,7 +16,7 @@ def image_classification(path, model):
predictions = model.predict(img_array) predictions = model.predict(img_array)
score = tf.nn.softmax(predictions[0]) score = tf.nn.softmax(predictions[0])
print(class_names[np.argmax(score)]) # print(class_names[np.argmax(score)])
return class_names[np.argmax(score)] return class_names[np.argmax(score)]