| # 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. |