3.2. 使用 Resnet-50 进行图像分类¶
在本节,我们将展示如何在 Sophon TPU 上面使用 resnet-50 进行图像分类。 我们使用的 bmodel 都是经过转换后的官方的 caffe 版本的 resnet-50,包括了 fp32 以及 int8 类型的。 我们实现了 4 个示例程序。 它们都是使用 opencv 进行图像解码和预处理。 图像都会被缩放到固定的尺寸,在这里是 1*3*224*224。 这4个程序的区别在于它们支持的模型数量、TPU 数量和线程数量不同。
ID |
Input |
Decoder |
Preprocessor |
Data Type |
Model |
Mode |
Model Number |
TPU Number |
Multi-Thread |
---|---|---|---|---|---|---|---|---|---|
0 |
image |
opencv |
opencv |
fp32 int8 |
resnet-50 |
static |
1 |
1 |
N |
1 |
image |
opencv |
opencv |
fp32 int8 |
resnet-50 |
static |
1 |
1 |
Y |
2 |
image |
opencv |
opencv |
fp32 int8 |
resnet-50 |
static |
2 |
1 |
Y |
3 |
image |
opencv |
opencv |
fp32 int8 |
resnet-50 |
static |
1 |
2 |
Y |