gpt4 book ai didi

opencv nms 学习笔记

转载 作者:知者 更新时间:2024-03-13 06:05:26 27 4
gpt4 key购买 nike

opencv nms

nanodet用到了:

nanodet onnx踩坑记录_jacke121的专栏-CSDN博客

indices = cv2.dnn.NMSBoxes(bboxes_wh.tolist(), confidences.tolist(), self.prob_threshold, self.iou_threshold)

倾斜四边形非极大值抑制(NMS)的计算思路

opencv的:

代码所在位置:

https://github.com/root12321/Rotation-Detect-yolov5_poly/blob/6ac68bd4900e3de8d70cc3089f93dd7a1dd56e1e/utils/general.py

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

这篇文章写的不错,挺详细的:

倾斜四边形非极大值抑制(NMS)的计算思路 - 简书

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com