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
]