fast nms YOLACT

源码地址：https://github.com/dbolya/yolact/blob/1722387d75210361c1f21c911d0e2420a48c7a23/layers/functions/detection.py

      def fast_nms(self, boxes, masks, scores, iou_threshold:float=0.5, top_k:int=200, second_threshold:bool=False):
       scores, idx = scores.sort(1, descending=True)
       idx = idx[:, :top_k].contiguous()
       scores = scores[:, :top_k]
       num_classes, num_dets = idx.size()
       boxes = boxes[idx.view(-1), :].view(num_classes, num_dets, 4)
       masks = masks[idx.view(-1), :].view(num_classes, num_dets, -1)
       iou = jaccard(boxes, boxes)
       iou.triu_(diagonal=1)
       iou_max, _ = iou.max(dim=1)
      # Now just filter out the ones higher than the threshold
       keep = (io
  
 

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

原文链接：blog.csdn.net/jacke121/article/details/92068855