27 lines
910 B
Markdown
27 lines
910 B
Markdown
|
# 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.
|