DOTA格式的8个坐标值转换为角度倾斜矩形框 (cx,cy,w,h,ang)

  

   

    

     

    
    

     

      def polygonToRotRectangle_batch(bbox, with_module=True):
     
    
   

    

     

    
    

     

          """
     
    
   

    

     

    
    

     

          :param bbox: The polygon stored in format [x1, y1, x2, y2, x3, y3, x4, y4]
     
    
   

    

     

    
    

     

                  shape [num_boxes, 8]
     
    
   

    

     

    
    

     

          :return: Rotated Rectangle in format [cx, cy, w, h, theta]
     
    
   

    

     

    
    

     

                  shape [num_rot_recs, 5]
     
    
   

    

     

    
    

     

          """
     
    
   

    

     

    
    

     

          # print('bbox: ', bbox)
     
    
   

    

     

    
    

     

          bbox = np.array(bbox,dtype=np.float32)
     
    
   

    

     

    
    

     

          bbox = np.reshape(bbox,newshape=(-1, 2, 4),order='F')
     
    
   

    

     

    
    

     

          # angle = math.atan2(-(bbox[0,1]-bbox[0,0]),bbox[1,1]-bbox[1,0])
     
    
   

    

     

    
    

     

          # print('bbox: ', bbox)
     
    
   

    

     

    
    

     

          angle = np.arctan2(-(bbox[:, 0,1]-bbox[:, 0,0]),bbox[:, 1,1]-bbox[:, 1,0])
     
    
   

    

     

    
    

     

          # angle = np.arctan2(-(bbox[:, 0,1]-bbox[:, 0,0]),bbox[:, 1,1]-bbox[:, 1,0])
     
    
   

    

     

    
    

     

          # center = [[0],[0]] ## shape [2, 1]
     
    
   

    

     

    
    

     

          # print('angle: ', angle)
     
    
   

    

     

    
    

     

          center = np.zeros((bbox.shape[0], 2, 1))
     
    
   

    

     

    
    

     

          for i in range(4):
     
    
   

    

     

    
    

     

              center[:, 0, 0] += bbox[:, 0,i]
     
    
   

    

     

    
    

     

              center[:, 1, 0] += bbox[:, 1,i]
     
    
   

    

     

    
    

     

       
     
    
   

    

     

    
    

     

          center = np.array(center,dtype=np.float32)/4.0
     
    
   

    

     

    
    

     

       
     
    
   

    

     

    
    

     

          # R = np.array([[math.cos(angle), -math.sin(angle)], [math.sin(angle), math.cos(angle)]], dtype=np.float32)
     
    
   

    

     

    
    

     

          R = np.array([[np.cos(angle), -np.sin(angle)], [np.sin(angle), np.cos(angle)]], dtype=np.float32)
     
    
   

    

     

    
    

     

       
     
    
   

    

     

    
    

     

          normalized = np.matmul(R.transpose((2, 1, 0)),bbox-center)
     
    
   

    

     

    
    

     

       
     
    
   

    

     

    
    

     

       
     
    
   

    

     

    
    

     

          xmin = np.min(normalized[:, 0, :], axis=1)
     
    
   

    

     

    
    

     

          # print('diff: ', (xmin - normalized[:, 0, 3]))
     
    
   

    

     

    
    

     

          # assert sum((abs(xmin - normalized[:, 0, 3])) > eps) == 0
     
    
   

    

     

    
    

     

          xmax = np.max(normalized[:, 0, :], axis=1)
     
    
   

    

     

    
    

     

          # assert sum(abs(xmax - normalized[:, 0, 1]) > eps) == 0
     
    
   

    

     

    
    

     

          # print('diff2: ', xmax - normalized[:, 0, 1])
     
    
   

    

     

    
    

     

          ymin = np.min(normalized[:, 1, :], axis=1)
     
    
   

    

     

    
    

     

          # assert sum(abs(ymin - normalized[:, 1, 3]) > eps) == 0
     
    
   

    

     

    
    

     

          # print('diff3: ', ymin - normalized[:, 1, 3])
     
    
   

    

     

    
    

     

          ymax = np.max(normalized[:, 1, :], axis=1)
     
    
   

    

     

    
    

     

          # assert sum(abs(ymax - normalized[:, 1, 1]) > eps) == 0
     
    
   

    

     

    
    

     

          # print('diff4: ', ymax - normalized[:, 1, 1])
     
    
   

    

     

    
    

     

       
     
    
   

    

     

    
    

     

          w = xmax - xmin + 1
     
    
   

    

     

    
    

     

          h = ymax - ymin + 1
     
    
   

    

     

    
    

     

       
     
    
   

    

     

    
    

     

          w = w[:, np.newaxis]
     
    
   

    

     

    
    

     

          h = h[:, np.newaxis]
     
    
   

    

     

    
    

     

          # TODO: check it
     
    
   

    

     

    
    

     

          if with_module:
     
    
   

    

     

    
    

     

              angle = angle[:, np.newaxis] % ( 2 * np.pi)
     
    
   

    

     

    
    

     

          else:
     
    
   

    

     

    
    

     

              angle = angle[:, np.newaxis]
     
    
   

    

     

    
    

     

          dboxes = np.concatenate((center[:, 0].astype(np.float), center[:, 1].astype(np.float), w, h, angle), axis=1)
     
    
   

    

     

    
    

     

          return dboxes
     
    
  
 

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

原文链接：wanghao.blog.csdn.net/article/details/117219250