s473603/SztIn_gr.234798
0
0
Fork 1
Code Issues 1 Pull Requests 1 Projects Releases Wiki Activity

master #1

Merged
s473554 merged 7 commits from s473554/SztIn_gr.234798:master into master 2023-03-23 13:05:47 +01:00
Conversation 0 Commits 7 Files Changed 2 -29
1 changed files with 0 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit a56373b33f - Show all commits

29
proverka.py
View File

@ -1,29 +0,0 @@
import os
import numpy as np
import tensorflow as tf
from tensorflow import keras
import cv2
directory = "C:/Users/KimD/PycharmProjects/Traktor_V1/Vegetable Images/test"
test_ds = tf.keras.utils.image_dataset_from_directory(directory, validation_split=0.2, image_size=(224, 224),
subset="validation", seed=123, batch_size=32)
model = keras.models.load_model("C:/Users/KimD/PycharmProjects/Traktor_V1/mode2.h5")
# predictions = model.predict(test_ds.take(32))
class_names = test_ds.class_names
img = cv2.imread('C:/Users/KimD/PycharmProjects/Traktor_V1/Vegetable Images/test/Carrot/1001.jpg')
cv2.imshow("lala", img)
cv2.waitKey(0)
img = (np.expand_dims(img, 0))
print(class_names)
predictions = model.predict(img)
print(predictions)