class importer.TensorflowLoader(model)

Translates a Tensorflow model to Acuity formats.

__init__(model)
Parameters:

model – The file path of the imported Tensorflow model file, a .pb file.

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

Load Tensorflow model.

Parameters:
  • inputs – Specify the input points of the Tensorflow graph, requries list type. Such as [‘input_point_1’, ‘input_point_2’].

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

  • outputs – Specify the output points of the Tensorflow graph, requries list type. For example, [‘output_point_1’, ‘output_point_2’].

Returns:

Acuity network.