opencv nms
nanodet用到了:
nanodet onnx踩坑记录_jacke121的专栏-CSDN博客
indices = cv2.dnn.NMSBoxes(bboxes_wh.tolist(), confidences.tolist(), self.prob_threshold, self.iou_threshold)
opencv的:
代码所在位置:
4.5.4版本 报错,我改成如下代码,不报错了。
i = cv2.dnn.NMSBoxesRotated(boxes_for_cv2_nms, scores_for_cv2_nms, conf_thres, iou_thres)
# i = np.squeeze(i, axis=-1)
if i.shape[0] > max_det: # limit detections
i = i[:max_det]
api:
def NMSBoxesRotated(bboxes,
scores,
score_threshold,
nms_threshold,
eta=None,
top_k=None):
pass
这篇文章写的不错,挺详细的:
版权说明 : 本文为转载文章, 版权归原作者所有 版权申明
原文链接 : https://blog.csdn.net/jacke121/article/details/123217086
内容来源于网络,如有侵权,请联系作者删除!