23 lines
582 B
Python
23 lines
582 B
Python
import numpy as np
|
|
import tensorflow as tf
|
|
from tensorflow import keras
|
|
|
|
|
|
# loaded_model = keras.models.load_model("my_model")
|
|
|
|
def image_classification(path, model):
|
|
class_names = ['door', 'refrigerator', 'shelf']
|
|
|
|
img = tf.keras.utils.load_img(
|
|
path, target_size=(180, 180)
|
|
)
|
|
img_array = tf.keras.utils.img_to_array(img)
|
|
img_array = tf.expand_dims(img_array, 0) # Create a batch
|
|
|
|
predictions = model.predict(img_array)
|
|
score = tf.nn.softmax(predictions[0])
|
|
# print(class_names[np.argmax(score)])
|
|
return class_names[np.argmax(score)]
|
|
|
|
|