File size: 471 Bytes
f33defa | 1 2 3 4 5 6 7 8 9 10 11 12 13 | """MobileNetV3-Small classification inference on AX650."""
import numpy as np
import axengine
class MobileNetV3Classifier:
def __init__(self, model_path: str):
self.sess = axengine.InferenceSession(model_path)
self.input_name = self.sess.get_inputs()[0].name
def classify(self, image: np.ndarray) -> np.ndarray:
"""image: float32 NCHW [1,3,224,224], range [0,1]"""
return self.sess.run(None, {self.input_name: image})[0]
|