File size: 1,133 Bytes
477da44
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
# 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
```
*rknn_convert usage:*
```
python3 -m rknn.api.rknn_convert -t rk3568 -i ./model_config.yml -o ./
```
*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] score:0.984375 class:"Shih-Tzu"
[ 155] score:0.007812 class:"Pekinese, Pekingese, Peke"
[ 205] score:0.003906 class:"Lhasa, Lhasa apso"
[  -1]: 0.0
[  -1]: 0.0
```
- Note: Different platforms, different versions of tools and drivers may have slightly different results.