昇腾应用案例体验：(10) 图像目标检测

昇腾AI应用，探索人工智能的无限可能，使能千行百业

图像目标检测

概述

该应用可部署至Atlas 200 DK 开发者套件（型号：3000）或者AI加速云服务器上，基于YOLOv3检测网络，通过读取本地图像数据作为输入，对图像中的物体进行目标检测，得到检测出的物体类别置信度以及检测目标的框，最后将检测的结果展示出来。

原理

1.运行管理资源申请：用于初始化系统内部资源，AscendCL接口固定的调用流程。

2.加载模型文件：从文件加载离线模型数据，需要由用户自行管理模型运行的内存，根据内存中加载的模型获取模型的基本信息包含模型输入、输出数据的数据buffer大小；由模型的基本信息构建模型输出内存，为接下来的模型推理做好准备。

3.获取本地图像并进行预处理：从本地存放有图片的目录中循环读取JPEG图像数据，使用DVPP的JPEGD功能进行图片解码，将jpg格式的图像数据解码为YUV420SP格式的图像，再使用DVPP的VPC功能将图像数据缩放至模型要求的宽高比例；然后构建模型的输入数据。

4.模型推理：根据构建好的模型输入数据进行模型推理。

5.解析推理结果：根据模型输出，解析目标检测的结果，得到图像数据中检测到的目标框，检测到的物体类别以及相似度。使用OpenCV将检测的结果标注在原始图像上并写到本地文件中。

技术

- 基于华为Ascend 310 AI处理器。

- 基于Caffe的YOLOv3目标检测模型，使用ATC模型转换工具将其转换为Ascend 310 AI处理器支持的模型。

硬件

Atlas 200 DK 开发者套件（型号：3000）

AI加速型云服务器

效果展示