Image Classification Model
Image classification introduction
Image classification is to identity what an image represents, to predict the object list and the probabilities. For example,the following table 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 |
|
15.0 |
73.9% |
91.40% |
- |
9.959 |
|
40.4 |
80.2% |
94.90% |
- |
52.243 |
|
15.3 |
73.6% |
- |
- |
31.452 |
|
17.8 |
93.7% |
- |
- |
9.082 |
|
48.6 |
80.2% |
- |
- |
89.816 |
|
97.3 |
95.4% |
- |
- |
63.227 |
|
80.5 |
95.0% |
- |
- |
20.652 |
|
89.6 |
94.5% |
- |
- |
24.561 |