介绍
目标检测(Object Detection)是计算机视觉领域的基本任务之一,学术界已有将近二十年的研究历史。近些年随着深度学习技术的火热发展,目标检测算法也从基于手工特征的传统算法转向了基于深度神经网络的检测技术。本应用提供了基于Atlas 200 DK 开发者套件(型号:3000)进行检测网络应用开发的Demo。
案例简介
本案例基于搭载了华为昇腾310芯片的A200DK/A300实现对yolov3网络的推理功能,快速实现输入原始mp4视频,使用yolov3进行检测标框并生成新的mp4视频输出。
原理
1.待检测视频:多线程样例支持输入多路视频。
2.数据预处理:在预处理子线程中,对解码出来的每一帧图像进行预处理,包括色域转换及图片resize等。处理完毕后将结果存入队列中。
3.模型推理:从队列中获取待处理的数据,根据构建好的模型输入数据进行模型推理。推理完成后将推理结果存到队列中。
4.解析推理结果:从队列中获取模型输出,根据模型输出,解析模型的推理结果。在对应的图片帧上标定检测框,并编码为视频文件。