class importer.TFLiteLoader(model)

Translates a TensorFlow Lite model to ACUITY formats. For quantized TensorFlow Lite models, do not quantize them again in Acuity. if set inputs param, need to use input_size_list parameter at the same time.

__init__(model)
Parameters:

model – The file path of the imported TensorFlow Lite (TFLite) model, in an optimized FlatBuffer format identified by the .tflite file extension.

load(inputs=None, input_size_list=None, outputs=None, **kwargs)

Load TFlite model.

Parameters:
  • inputs – Specify the input points of the TFLite graph, requries list type. Such as [‘input_point_1’, ‘input_point_2’]. When this argument is omitted, Acuity uses all header points of this model.

  • input_size_list – The tensor shape sizes of the input points listed in the inputs argument, requires list type. For example, [[1,3,224,224],[1,12,1]], which represents the tensor shape sizes of two input points.

  • outputs – Specify the output points of the TFLite graph, requries list type. For example, [‘output_point_1’, ‘output_point_2’]. When this argument is omitted, Acuity uses all tail points of this model.

Returns:

Acuity network.