opencv nms 学习笔记

举报
风吹稻花香 发表于 2022/03/02 23:11:44 2022/03/02
2.1k+ 0 0
【摘要】 opencv nms nanodet用到了: nanodet onnx踩坑记录_jacke121的专栏-CSDN博客 indices = cv2.dnn.NMSBoxes(bboxes_wh.tolist(), confidences.tolist(), self.prob_threshold, self.iou_thresh...

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)的计算思路 - 简书

文章来源: blog.csdn.net,作者:AI视觉网奇,版权归原作者所有,如需转载,请联系作者。

原文链接:blog.csdn.net/jacke121/article/details/123217086

【版权声明】本文为华为云社区用户转载文章,如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

作者其他文章

评论(0

抱歉,系统识别当前为高风险访问,暂不支持该操作

    全部回复

    上滑加载中

    设置昵称

    在此一键设置昵称,即可参与社区互动!

    *长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

    *长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。