Image Classification Model Support (Lite)
Image classification introduction
Image classification is to identity what an image represents, to predict the object list and the probabilites. For example，the following tabel shows the classification results after mode inference.
Category |
Probability |
|---|---|
plant |
0.9359 |
flower |
0.8641 |
tree |
0.8584 |
houseplant |
0.7867 |
Using MindSpore Lite to realize image classification example.
Image classification model list
The following table shows the data of some image classification models using MindSpore Lite inference.
The performance of the table below is tested on the mate30.
Model name |
Size(Mb) |
Top1 |
Top5 |
F1 |
CPU 4 thread delay (ms) |
|---|---|---|---|---|---|
11.5 |
- |
- |
65.5% |
14.595 |
|
90.9 |
78.62% |
94.08% |
- |
92.086 |
|
8.8 |
67.74% |
87.62% |
- |
8.303 |
|
25.3 |
72.2% |
90.06% |
- |
23.257 |
|
95.8 |
73.1% |
91.21% |
- |
138.164 |