Method _evaluateTFLiteModel

tensorflow/lite/python/lite_v2_test.py:44–55

Evaluates the model on the `input_data`.

(self, tflite_model, input_data)

class TestModels(test_util.TensorFlowTestCase):

  def _evaluateTFLiteModel(self, tflite_model, input_data):
    """Evaluates the model on the `input_data`."""
    interpreter = Interpreter(model_content=tflite_model)
    interpreter.allocate_tensors()

    input_details = interpreter.get_input_details()
    output_details = interpreter.get_output_details()

    for input_tensor, tensor_data in zip(input_details, input_data):
      interpreter.set_tensor(input_tensor['index'], tensor_data.numpy())
    interpreter.invoke()
    return interpreter.get_tensor(output_details[0]['index'])

  def _getSimpleVariableModel(self):
    root = tracking.AutoTrackable()