/tflitego_examples

tflitego_examples provide examples to use tflitego to use TensorFlow lite in Go. More information about the library: https://github.com/nbortolotti/tflitego

Primary LanguageGoApache License 2.0Apache-2.0

Subreddit subscribers

Examples for tflitego

how to use

go get github.com/nbortolotti/tflitego

Examples

Iris Model

This example propose an inference using tflitego.

included:

  • iris.go: all the example code to implement an inference using tflitego
  • iris_lite.tflite: tflite generic model to run the example.

Image Categorization

This example propose inferences using images and with the objective to provide categorization or labeling.

included:

  • image.go:
  • image_test.go
  • images: folder with image placerholders.
  • models: folder with tf models.
    • mobilenet_v1_1.0_224_quant.tflite
    • mobilenet_v2_1.0_224_quant.tflite
    • labels_mobilenet_224.txt

Note: ioncluded a flag to customize an image capture into Raspberry Pi device. --rasp=true. This option provide th eoportunity to use a Raspberry Pi Camera Module to provide an interative demostration of image categorization.

Note:

remember to navigate into examples folder.

go build
go run iris.go