图像分类模型支持(Lite)
图像分类介绍
图像分类模型可以预测图片中出现哪些物体,识别出图片中出现物体列表及其概率。 比如下图经过模型推理的分类结果为下表:
类别 |
概率 |
---|---|
plant |
0.9359 |
flower |
0.8641 |
tree |
0.8584 |
houseplant |
0.7867 |
使用MindSpore Lite实现图像分类的示例代码。
图像分类模型列表
下表是使用MindSpore Lite推理的部分图像分类模型的数据。
下表的性能是在mate30手机上测试的。
模型名称 |
大小(Mb) |
Top1 |
Top5 |
F1 |
CPU 4线程时延(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 |