torchvision nms

要求格式[x1,y1,x2,y2]

  

   

    

     

    
    

     

      if __name__ == '__main__':
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 import torch
     
    
   

    

     

    
    

     

       a=torch.Tensor([[1,1,2,2],[1,1,3.100001,3],[1,1,3.1,3]])
     
    
   

    

     

    
    

     

       b=torch.Tensor([[0.9],[0.98],[0.980005]])
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 from torchvision.ops import nms
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       ccc=nms(a,b,0.4)
     
    
   

    

     

    
    

     

       print(ccc)
     
    
   

    

     

    
    

     

       print(a[ccc])
     
    
  
 

  

   

    

     

    
    

     

      from __future__ import print_function
     
    
   

    

     

    
    

     

      import os
     
    
   

    

     

    
    

     

      import argparse
     
    
   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     

      import torch.backends.cudnn as cudnn
     
    
   

    

     

    
    

     

      import numpy as np
     
    
   

    

     

    
    

     

      from torchvision.ops import nms
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      from data import cfg_mnet, cfg_re50, cfg_peleenet
     
    
   

    

     

    
    

     

      from layers.functions.prior_box import PriorBox
     
    
   

    

     

    
    

     

      from utils.nms.py_cpu_nms import py_cpu_nms
     
    
   

    

     

    
    

     

      import cv2
     
    
   

    

     

    
    

     

      from models.retinaface import RetinaFace
     
    
   

    

     

    
    

     

      from utils.box_utils import decode, decode_landm
     
    
   

    

     

    
    

     

      import time
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      parser = argparse.ArgumentParser(description='Retinaface')
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      # parser.add_argument('-m', '--trained_model', default='weights/mobilenet0.25_Final.p
     
    
  
 

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

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