Giter VIP home page Giter VIP logo

qengineering / tensorflow_lite_classification_rpi_zero Goto Github PK

View Code? Open in Web Editor NEW
9.0 2.0 1.0 856 KB

TensorFlow Lite on a bare Raspberry Pi Zero

Home Page: https://qengineering.eu/install-tensorflow-2-lite-on-raspberry-pi-4.html

License: BSD 3-Clause "New" or "Revised" License

C++ 100.00%
deep-learning raspberry-pi-zero-w raspberry-pi-zero tensoflow-lite mobilenet armv6 cpp tensorflow-examples

tensorflow_lite_classification_rpi_zero's Introduction

TensorFlow_Lite_Classification_RPi_zero

output image

TensorFlow Lite classification running on a bare Raspberry Pi Zero

License

A 'fast' C++ implementation of TensorFlow Lite classification on a bare Raspberry Pi zero.
Be noted that we use the zero version here, not the new Raspberry Pi zero 2.
Inference time: 11 sec
Special made for a Jetson Nano see Q-engineering deep learning examples


Papers: https://arxiv.org/pdf/1712.05877.pdf
Training set: COCO with 1000 objects
Size: 224x224


Dependencies.

To run the application, you have to:


Installing the app.

To extract and run the network in Code::Blocks
$ mkdir MyDir
$ cd MyDir
$ wget https://github.com/Qengineering/TensorFlow_Lite_Classification_RPi_zero/archive/refs/heads/master.zip
$ unzip -j master.zip
Remove master.zip and README.md as they are no longer needed.
$ rm master.zip
$ rm README.md

Your MyDir folder must now look like this:
tabby.jpeg
schoolbus.jpg
grace_hopper.bmp
Labels.txt
TensorFlow_Lite_Mobile.cpb
TensorFlow_Lite_Class.cpp

Next, choose your model from TensorFlow: https://www.tensorflow.org/lite/guide/hosted_models
Download a quantized model, extract the .tflite from the tarball and place it in your MyDir.

Now your MyDir folder may contain: mobilenet_v1_1.0_224_quant.tflite.
Or: inception_v4_299_quant.tflite. Or both of course.

Enter the .tflite file of your choice on line 54 in TensorFlow_Lite_Class.cpp
The image to be tested is given a line 84, also in TensorFlow_Lite_Class.cpp


Running the app.

Run TestTensorFlow_Lite.cpb with Code::Blocks. More info or
if you want to connect a camera to the app, follow the instructions at Hands-On.


paypal

tensorflow_lite_classification_rpi_zero's People

Contributors

qengineering avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar

tensorflow_lite_classification_rpi_zero's Issues

Federated Learning on ARM64

Hi Sir,
Are you also working on federated learning. Could you please share something on classification using federated learning on ARM64 devices please?

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.