Faster R-CNN:如何避免在同一区域进行多次检测?
机器算法验证
神经网络
卷积神经网络
张量流
物体检测
2022-04-11 13:05:19
2个回答
这是对象检测器(例如 Faster R-CNN)的一个共同属性:它们多次预测每个对象。过滤掉重复项是非最大抑制函数的工作。松散地解释,NMS 采用一对具有相同类别的重叠框,如果它们的重叠大于某个阈值,则只保留概率较高的框。此过程继续进行,直到没有更多具有足够重叠的框为止。这个最小重叠率是您可以调整的超参数之一。
您可以调整的第二个超参数是类别概率的阈值(例如 70%)。所有以较低概率预测的对象都被简单地忽略。
调整这两个超参数应该会给你一个令人满意的预测质量。
如前所述,NMS 用于去除误报。由于您使用的是更快的 R-CNN,因此 NMS 的阈值自动设置为 0.7。因此,您可以使用 OpenCV 函数 cv2.dnn.NMSBoxes(boxes, confidences, confid, thresh) 您可以深入研究它以获取更多信息。
其它你可能感兴趣的问题

