android13/hardware/rockchip/rknn-toolkit2/examples/tflite/mobilenet_v1_qat

TFLite MobileNet V1 QAT

Model Source

The model used in this example come from the following open source projects:
https://www.tensorflow.org/lite/examples/image_classification/overview?hl=zh-cn

Script Usage

Usage:

python test.py

Description:

• The default target platform in script is 'rk3566', please modify the 'target_platform' parameter of 'rknn.config' according to the actual platform.
• If connecting board is required, please add the 'target' parameter in 'rknn.init_runtime'.
• This is a QAT model, and the do_quantization of rknn.build needs to be set to False.

Expected Results

This example will print the TOP5 labels and corresponding scores of the test image classification results, as follows:

-----TOP 5-----
[156]: 252
[155]: 3
[205]: 1
-1: 0.0
-1: 0.0

• Note: Different platforms, different versions of tools and drivers may have slightly different results.