Views
No views yet
1import onnxruntime as ort, cv2, numpy as np
2session = ort.InferenceSession("model_age.onnx")
3inp = session.get_inputs()[0].name
4out = session.get_outputs()[0].name
5img = cv2.imread("test.jpg")
6img = cv2.resize(img, (224,224)).astype(np.float32)[None,:,:,:]
7pred = session.run([out], {inp: img})