一种基于瑞芯微平台的轻量化改进目标检测方法和装置

摘要

本发明公开了一种基于瑞芯微平台的轻量化改进目标检测方法和装置。模型以Tinyyolov4为算法基础，利用k‑means算法对自建数据集进行先验框改进；根据移植部署的瑞芯微平台硬件单元特点对骨干网络中的卷积输出通道数、激活函数均进行改进；通过非对称8位量化优化转换方式获得一种轻量化的目标检测模型。装置以改进后算法为核心，利用RKNN工具链转换为可部署在瑞芯微平台上的.rknn模型；通过CPU调度NPU加载.rknn模型，并加速推理经GPU预处理的图像数据，实现前向推理；CPU调度GPU进行数据后处理，处理后的数据传输到显示器上显示。本发明的方法和装置实现了：轻量化后的目标检测模型在轻微损失准确率时，权重参数量减小和检测速度增快；可以实现实时鲁棒的图像检测。