昇腾社区首页
EN
注册
后处理插件开发教程

概述

本应用以昇腾 Atlas 300I pro、Atlas 300V pro 为主要的硬件平台,对于多任务模型,需要按照插件的形式开发后处理。以yolov3为例,开发yolov3后处理插件。

功能模块介绍

本案例流程如下:

预处理:图像解码和图像缩放等前处理操作,满足模型的输入尺寸

yolov3模型推理:目标检测模型推理,输出3层特征图,需要通过后处理解码出来坐标框

自定义yolov3后处理插件:不使用SDK自带的目标检测插件,独立完成自定义后处理插件开发

效果展示