dataeval.models.LiteRtImageClassifier¶
-
class dataeval.models.LiteRtImageClassifier(model_path, metadata_path, *, image_size=
None, scores_name='scores')¶ Opinionated image classifier backed by LiteRT (TensorFlow Lite).
Loads an exported
.tfliteimage-classification model via a LiteRT interpreter together with its model-metadata.json contract, then maps a batch of CHW images to per-image class scores. Calling the instance preprocesses each image to the model’s input contract (color layout, size,[0, 1]normalization), runs inference, and returns one(nClasses,)float32 score array per image. Requiresdataeval[tflite].- Parameters:¶
- model_path : str or Path¶
Path to the exported
.tflitemodel file.- metadata_path : str or Path¶
Path to the model-metadata.json describing the input/output contract. Its declared task must be
IMAGE_CLASSIFICATION.- image_size : tuple[int, int] or None, default None¶
Optional
(height, width)override for the model’s input size. When set, images are resized to this size instead of the size declared in metadata; required when the metadata declares a variable (-1) dimension.- scores_name : str, default "scores"¶
Name of the model output tensor holding class scores.
- Raises:¶
ValueError – If the metadata declares a task other than
IMAGE_CLASSIFICATION.ImportError – If neither
tflite-runtimenortensorflowis installed.FileNotFoundError – If
model_pathdoes not exist.
See also
OnnxImageClassifierSame classifier backed by ONNX Runtime.
LiteRtObjectDetectorOpinionated object detector backed by LiteRT.
read_model_metadataParse the metadata contract.
Notes
Implements the MAITE
image_classification.Modelprotocol: instances expose aModelMetadatametadataattribute and are callable on a batch of images.