目标检测旋转增强源码

解法1：

  

   

    

     

    
    

     

      import cv2
     
    
   

    

     

    
    

     

      import math
     
    
   

    

     

    
    

     

      import numpy as np
     
    
   

    

     

    
    

     

      import os
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def get_degree(p1,p0):
     
    
   

    

     

    
    

     

       aaa = math.degrees(math.atan2((p1[1] - p0[1]), (p1[0] - p0[0])))
     
    
   

    

     

    
    

     
 if aaa < 0:
     
    
   

    

     

    
    

     

       aaa = 360 + aaa
     
    
   

    

     

    
    

     

       aaa = (aaa + 90)
     
    
   

    

     

    
    

     
 if aaa > 360:
     
    
   

    

     

    
    

     

       aaa = aaa - 360
     
    
   

    

     

    
    

     
 return aaa
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def rotate_image(src,box_points, angle, scale=1.):
     
    
   

    

     

    
    

     

       nw = src.shape[1]
     
    
   

    

     

    
    

     

       nh = src.shape[0]
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       rot_mat = cv2.getRotationMatrix2D((nw*0.5, nh*0.5), angle, scale)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       rot_move = np.dot(rot_mat, np.array([0, 0,0]))
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       rot_mat[0,2] += rot_move[0]
     
    
   

    

     

    
    

     

       rot_mat[1,2] += rot_move[1]
     
    
   

    

     

    
    

     

       new_img=cv2.warpAffine(src, rot_mat, (int(math.ceil(nw)),
     
    
  
 

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

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