Toybrick

标题: 使用yolov3_tiny转换的rknn模型检测后，结果框重复 [打印本页]

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

作者: jefferyzhang 时间: 2020-4-2 11:17
做NMS啊

作者: double_s 时间: 2020-4-2 12:00

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

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

作者: double_s 时间: 2020-4-2 16:52

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

好的，谢谢

欢迎光临 Toybrick (https://t.rock-chips.com/)