使用yolov3_tiny转换的rknn模型检测后，结果框重复

[复制链接] · 发表于 2020-4-2 11:16:08

大家好，我使用rknn_transform_tiny.py将yolov3_tiny的模型转为rknn模型后，用rknn_picture.py程序进行检测，确认参数设置是正确的，mask部分也使用的tiny的。但是在检测的结果中，同一个目标常出现几个不同的框，类似下图：（每张检测图都会有不同程度的重复框）在原版darknet中使用yolov3_tiny检测的结果中，同一个目标只有一个框，请问是什么原因呢？针对tiny版的rknn_picture.py代码有哪里需要进行更改吗？

只看该作者 · 发表于 2020-4-2 11:17:58

做NMS啊

只看该作者 · 发表于 2020-4-2 12:00:13

jefferyzhang 发表于 2020-4-2 11:17
做NMS啊

做了呀~程序中不是有nms处理那步吗，nms_boxes这个函数

只看该作者 · 发表于 2020-4-2 16:52:38

hisping 发表于 2020-4-2 15:12
把NMS_THRESH改小就行了，IOU = 重叠面积 / (框1面积+框2面积-重叠面积)

好的，谢谢

使用yolov3_tiny转换的rknn模型检测后，结果框重复

本帖子中包含更多资源