hand_ges / predict.py
Manar312's picture
Upload predict.py
5b3ea43 verified
Raw
History Blame Contribute Delete
415 Bytes
import tensorflow as tf
import numpy as np
from preprocessing import preprocess
from labels import classes
model = tf.keras.models.load_model("hand_gesture_Model.h5")
def predict_image(image):
image = preprocess(image)
prediction = model.predict(image)[0]
results = {}
for i,prob in enumerate(prediction):
results[classes[i]] = float(prob)
return results