tosanoob/mediapipe_fastapi_20classes

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