pytorch 批量 iou

批量max

  

   

    

     

    
    

     

      a = torch.Tensor([[random.randint(0, 20), random.randint(0, 20), random.randint(0, 20)]])
     
    
   

    

     

    
    

     

      b1_x1=torch.Tensor([[random.randint(0, 20),random.randint(0, 20), random.randint(0, 20)]])
     
    
   

    

     

    
    

     

      inter_rect_x1 = torch.max(b1_x1, a)
     
    
   

    

     

    
    

     

      print(a)
     
    
   

    

     

    
    

     

      print(b1_x1)
     
    
   

    

     

    
    

     

      print(inter_rect_x1)
     
    
  
 

批量iou

  

   

    

     

    
    

     

      #!/usr/bin/env python
     
    
   

    

     

    
    

     

      # -*- coding: utf-8 -*-
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import datetime
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     

      from numba import jit
     
    
   

    

     

    
    

     

      import numpy as np
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import random
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      #面积计算法，前提，坐标都需要转成int类型
     
    
   

    

     

    
    

     

      def bbox_iou2(box1, box2, x1y1x2y2=True):
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       a= np.zeros((6,6),dtype=np.int)
     
    
   

    

     

    
    

     

       a[box1[:,1][0]:box1[:,3][0]+1,box1[:,0][0]:box1[:,2][0]+1]=1
     
    
   

    

     

    
    

     

       a[box2[:,1][0]:box2[:,3][0]+1, box2[:,0][0]:box2[:,2][0]+1] += 1
     
    
   

    

     

    
    

     

       m = np.sum(a[np.where(a ==1)])
     
    
   

    

     

    
    

     

       inner = np.sum(a[np.where(a == 2)])//2
     
    
   

    

     

    
    

     
 return inner/(inner+m)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def bbox_iou(box
     
    
  
 

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

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