Check a prototype of 2nd version of this platform being developed by cKnowledge.org in collaboration with MLCommons.
docker:image-classification-tflite.debian-9 (v3.0.0)
Share
Copyright: See copyright in the source repository
License: See license in the source repository
Creation date: 2019-05-17
Source: GitHub
cID: 88eef0cd8c43b68a:a0b9e9441cea153b

Don't hesitate to get in touch if you encounter any issues or would like to discuss this community project!
Please report if this CK component works: 1 ▲ or fails: 0 ▲
Sign up to be notified when artifacts are shared or updated!

Description  

This CK-powered container is our attempt to provide a common API to customize, build and run AI and ML applications with different models, frameworks, libraries, datasets, compilers, formats, backends and platforms. Our on-going project is to make the onboarding process as simple as possible via this platform. Please check this CK white paper and don't hesitate to contact us if you have suggestions or feedback!

ReadMe  

MLPerf Inference - Image Classification - TFLite (Debian 9)

  1. Default image (based on Debian 9 latest)

NB: You may need to run commands below with sudo, unless you manage Docker as a non-root user.

Default image

Download

$ docker pull ctuning/image-classification-tflite.debian-9

Build

$ ck build docker:image-classification-tflite.debian-9

NB: Equivalent to:

$ cd `ck find docker:image-classification-tflite.debian-9`
$ docker build -f Dockerfile -t ctuning/image-classification-tflite.debian-9 .

Run

Image Classification (default command)

$ ck run docker:image-classification-tflite.debian-9

NB: Equivalent to:

$ docker run --rm ctuning/image-classification-tflite.debian-9 \
"ck run program:image-classification-tflite --dep_add_tags.weights=mobilenet,non-quantized --env.CK_BATCH_COUNT=2"

Image Classification (custom command)

ResNet
$ docker run --rm ctuning/image-classification-tflite.debian-9 \
"ck run program:image-classification-tflite --dep_add_tags.weights=resnet,no-argmax --env.CK_BATCH_COUNT=500"
...
Summary:
-------------------------------
Graph loaded in 0.001373s
All images loaded in 1.173109s
All images classified in 83.794106s
Average classification time: 0.167370s
Accuracy top 1: 0.762 (381 of 500)
Accuracy top 5: 0.93 (465 of 500)
--------------------------------
MobileNet non-quantized
$ docker run --rm ctuning/image-classification-tflite.debian-9 \
"ck run program:image-classification-tflite --dep_add_tags.weights=mobilenet,non-quantized --env.CK_BATCH_COUNT=500"
...
Summary:
-------------------------------
Graph loaded in 0.001232s
All images loaded in 0.688655s
All images classified in 27.330872s
Average classification time: 0.054614s
Accuracy top 1: 0.724 (362 of 500)
Accuracy top 5: 0.896 (448 of 500)
--------------------------------
MobileNet quantized
$ docker run --rm ctuning/image-classification-tflite.debian-9 \
"ck run program:image-classification-tflite --dep_add_tags.weights=mobilenet,quantized --env.CK_BATCH_COUNT=500"
...
Summary:
-------------------------------
Graph loaded in 0.001149s
All images loaded in 0.032835s
All images classified in 81.675827s
Average classification time: 0.163378s
Accuracy top 1: 0.728 (364 of 500)
Accuracy top 5: 0.898 (449 of 500)
--------------------------------

Bash

$ docker run -it --rm ctuning/image-classification-tflite.debian-9 bash

Versions  

Files  

Comments  

Please log in to add your comments!
If you notice any inapropriate content that should not be here, please report us as soon as possible and we will try to remove it within 48 hours!