ctuning/image-classification-tflite.centos-7 repository overview

This container may be outdated. Please check CK-MLPerf docs⁠.

This repository contains Docker-ized instructions for the Collective Knowledge-powered image classification TFLite workloads in the MLPerf Inference benchmark: MobileNet (non-quantized, quantized), ResNet.

After pulling the latest image:

$ docker pull ctuning/image-classification-tflite.centos-7

you can use it e.g. as follows:

• run image classification using non-quantized MobileNet on a single image:

$ docker run --rm ctuning/image-classification-tflite.centos-7 \
"ck run program:image-classification-tflite --dep_add_tags.weights=mobilenet,non-quantized"

• run image classification using ResNet without the ArgMax operator on 10 images:

$ docker run --rm ctuning/image-classification-tflite.centos-7 \
"ck run program:image-classification-tflite --dep_add_tags.weights=resnet,no-argmax --env.CK_BATCH_COUNT=10"

• step into bash:

$ docker run -it --rm ctuning/image-classification-tflite.centos-7 bash

For further information, see:

• build and run instructions⁠;
• benchmarking instructions⁠.