概述
本应用以昇腾Atlas 300I为主要的硬件平台,使用Vision SDK实现FCOS目标检测的功能
功能模块介绍
1. 将待检测的图片放到相应的文件夹下。
2. 首先使用mmcv库对图片进行前处理(改变图片大小、归一化、补边操作)。
3. 将图片转换成为二进制的numpy数组,传入pipeline中。
4. 放缩后的图片输入模型推理插件mxpi_tensorinfer进行处理;
5. 将经过模型推理输出时候的张量数据流输入到mxpi_objectpostprocessor中,对FCOS目标检测模型推理输出的张量进行后处理;
6. 处理完毕之后将数据传入mxpi_dataserialize0插件中,将stream结果组装成json字符串输出。