yolov5 rknn后处理
时间: 2023-10-24 10:06:38 浏览: 120
yolov5-7.0实例分割模型转onnx再转rknn,含各个阶段代码
yolov5 rknn后处理主要包括以下几个步骤:
1. 解析rknn输出:rknn模型输出的是一个一维数组,需要根据模型输出的shape和anchor等信息解析出每个检测框的位置、置信度和类别概率等信息。
2. 进行非极大值抑制(NMS):对于同一物体可能会被多个检测框检测到的情况,需要进行NMS处理,保留置信度最高的检测框。
3. 根据anchor还原检测框:在进行NMS处理后,需要根据anchor还原检测框的位置。
4. 进行后处理:包括将检测框坐标从归一化坐标还原到原图坐标、筛选出置信度高于阈值的检测框、根据类别概率进行筛选等。
阅读全文