A simple inference API for the post-training dynamic range model quantization

Post-training_dynamic_range_quantization--inference_API--03.py

class TfLiteModel():
    def __init__(self, model):
        self.model = model
        
    def predict(self, X, **kwargs):
        _tflite_model = dynamic_range_quantize_model(self.model)
        _x = np.asarray(X); assert len(X.shape)>=2
        _tflite_out_preds = list(map(lambda i : tflite_preds(np.expand_dims(_x[i,:],axis=0), _tflite_model), range(_x.shape[0]) ))
        return np.asarray(_tflite_out_preds)