Models¶
spokestack.models.tensorflow module¶
TFLite model base class
-
class
spokestack.models.tensorflow.
TFLiteModel
(model_path, **kwargs)[source]¶ -
TFLite model base class for managing multiple inputs/outputs
- Parameters
-
-
model_path (str) – Path to .tflite model file
-
**kwargs (Any) – Additional keywords arguments for the TFLite Interpreter. [https://www.tensorflow.org/api_docs/python/tf/lite/Interpreter]
-
-
property
input_details
¶ -
Property for accesing the TFLite model input_details
Returns: Input details for the TFLite model
- Return type
-
List
[Any
]
-
property
output_details
¶ -
Property for accesing the TFLite model output_details
Returns: Output details for the TFLite model
- Return type
-
List
[Any
]