Skip to content

mattiantonini/edgetpu-compiler-container

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 

Repository files navigation

EdgeTPU Compiler Docker container

This project allows you to convert quantized TFLite models in order to execute them on the Coral Accelerator or Coral Dev Board even if you do not have the EdgeTPU Compiler installed on your machine or if it is not supported like on MacOs X.

Requirements

The only requirement is a Docker runtime on your machine. You can install it by following this link!

How to install EdgeTPU Compiler with Docker

  1. Make sure you installed Docker by typing
$ docker version

This container has been tested with

  • Docker CE 18.09.0 on Ubuntu 16.04.1.
  • Docker CE 19.03.1 on MacOs X 10.14.5.
  1. Clone this repository
$ git clone git@github.com:mattiantonini/edgetpu-compiler-container.git
  1. Enter in the directory
$ cd edgetpu-compiler-container
  1. Build the container
$ docker build -t edgetpu-compiler .

This step will take a few minutes.

  1. Run the container and convert a model
$ docker run --rm -it -v $(pwd):/home/edgetpu-compiler edgetpu-compiler edgetpu_compiler <MODEL>_quant.tflite

This command converts the file <MODEL>_quant.tflite in the current directory to <MODEL>_quant_edgetpu.tflite that can be executed on the EdgeTPU. A working example is the following

$ docker run --rm -it -v $(pwd):/home/edgetpu-compiler edgetpu-compiler edgetpu_compiler squeezenet_quant.tflite

About

Docker container for Google EdgeTPU compiler

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published