55次浏览
本应用以昇腾 Atlas 300I pro、Atlas 300V pro 为主要的硬件平台,对于多任务模型,需要按照插件的形式开发后处理。以yolov3为例,开发yolov3后处理插件。
本案例流程如下:
预处理:图像解码和图像缩放等前处理操作,满足模型的输入尺寸
yolov3模型推理:目标检测模型推理,输出3层特征图,需要通过后处理解码出来坐标框
自定义yolov3后处理插件:不使用SDK自带的目标检测插件,独立完成自定义后处理插件开发
无