3.3. 使用 SSD 进行目标检测

在本节,我们将展示如何在 Sophon TPU 上面使用 ssd300-vgg16 进行目标检测。 我们使用的 bmodel 都是经过转换后的官方的 caffe 版本的 ssd300-vgg16,包括了 fp32 以及 int8 类型的。 我们实现了 5 个示例程序。 其中,case 0,case 1 的区别在于解码方式与数据预处理的方式不同。 case 2 是 case 1 的多 btach 版本,可以体现 int8 模型在 batch 为 4 的倍数时较优异的性能。

在本节的示例程序中,我们将重点介绍图像/视频的解码和预处理。 关于模型推理部分,即 sail::Engine 的使用,请参考 “2.2 使用 Resnet-50 进行图像分类” 中的相关说明。

ID

Input

Decoder

Preprocessor

Data Type

Model

Mode

Model Number

Batch Size

Multi-Thread

SOC Only

0

video image

opencv

opencv

fp32 int8

ssd300-vgg16

static

1

1

N

N

1

video image

bm-ffmpeg

bmcv

fp32 int8

ssd300-vgg16

static

1

1

N

N

2

video image

bm-ffmpeg

bmcv

fp32 int8

ssd300-vgg16

static

1

4

N

N

3

video image

bm-opencv

bm-opencv

fp32 int8

ssd300-vgg16

static

1

1

N

Y

4

video image

bm-opencv

bmcv

fp32 int8

ssd300-vgg16

static

1

1

N

Y