Skip to main content

A Python package to simplify the deployment process of exported

Project description

Teachable Machine Lite

MIT License Downloads PyPI

Description

A Python package to simplify the deployment process of exported Teachable Machine models into different Robotics, AI and IoT controllers such as: Raspberry Pi, Jetson Nano and any other SBCs using TensorFlowLite framework.

Developed by @MeqdadDev

Supported Classifiers

Image Classification: use exported and quantized TensorFlow Lite model from Teachable Machine platfrom (a model file with tflite extension).

Requirements

Python >= 3.7

How to install package

pip install teachable-machine-lite

Dependencies

numpy
tflite-runtime
Pillow (PIL)

How to Use Teachable Machine Lite Package

from teachable_machine_lite import TeachableMachineLite
import cv2 as cv

cap = cv.VideoCapture(0)

model_path = 'model.tflite'
image_file_name = "frame.jpg"
labels_path = "labels.txt"

tm_model = TeachableMachineLite(model_path=model_path, labels_file_path=labels_path)

while True:
    ret, frame = cap.read()
    cv.imshow('Cam', frame)
    cv.imwrite(image_file_name, frame)
    
    results = tm_model.classify_frame(image_file_name)
    print("results:",results)
    
    k = cv.waitKey(1)
    if k% 255 == 27:
        # press ESC to close camera view.
        break

Links:

PyPI

Source Code

Developer

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

teachable-machine-lite-1.1.tar.gz (5.0 kB view hashes)

Uploaded Source

Built Distribution

teachable_machine_lite-1.1-py3-none-any.whl (5.1 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page