This library provides a simple interface for image classification using TensorFlow Lite models. It's designed to work with pre-trained models and can process both single images and directories of images.
Installation
pipinstallimBroker
Features
Single image classification
Batch classification for directories
Support for custom TFLite models
Handles any type of image shapes
Usage
Initializing the Broker
fromimBrokerimporttflBroker# Define your TFlite model's pathmodel_path="path/to/your/model.tflite"# Define your output labelsoutput_labels={0:'Label 1',1:'Label 2',...}# Initialize the brokerbroker=tflBroker(model_path,output_labels)