diou ciou torch

diou比ciou要小，比iou的值也小，训练时候用比较好。

测试时还是iou比较好,训练时giou比较好

  

   

    

     

    
    

     

      import math
     
    
   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def bbox_iou(box1, box2, x1y1x2y2=True, GIoU=False, DIoU=False, CIoU=False):
     
    
   

    

     

    
    

     
 # Returns the IoU of box1 to box2. box1 is 4, box2 is nx4
     
    
   

    

     

    
    

     

       box2 = box2.t()
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 # Get the coordinates of bounding boxes
     
    
   

    

     

    
    

     
 if x1y1x2y2:  # x1, y1, x2, y2 = box1
     
    
   

    

     

    
    

     

       b1_x1, b1_y1, b1_x2, b1_y2 = box1[:, 0], box1[:, 1], box1[:, 2], box1[:, 3]
     
    
   

    

     

    
    

     

       b2_x1, b2_y1, b2_x2, b2_y2 = box2[:, 0], box2[:, 1], box2[:, 2], box2[:, 3]
     
    
   

    

     

    
    

     
 else:  # x, y, w, h = box1
     
    
   

    

     

    
    

     

       b1_x1, b1_x2 = box1[:, 0] - box1[:, 2] / 2, box1[:, 0] + box1[:, 2] / 2
     
    
   

    

     

    
    

     

       b1_y1, b1_y2 = box1[:, 1] - box1[:, 3] / 2, box1[:, 1] + box1[:, 3] / 2
     
    
   

    

     

    
    

     

       b2_x1, b2_x2 = box2[:, 0] - box2[:, 2] / 2, box2[:, 0] + box2[:, 2] / 2
     
    
   

    

     

    
    

     

       b2_y1, b2_y2 = box2[:, 1] - box2[:, 3] / 2, box2[:, 1] + box2[:, 3] / 2
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 # I
     
    
  
 

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

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