File size: 411 Bytes
aa25f09 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
from model import load_model
import json
import numpy as np
classifier = load_model()
with open("label_list.json","r") as infile:
actions = list(json.load(infile).values())
def model_predict(input):
"""Perform prediction on input (numpy array), return a label (str)"""
res = classifier.predict(np.expand_dims(input,axis=0))[0]
label = actions[np.argmax(res)]
return label |