python 矩形补正方形

roi补正方形：

boxes结构：x1 y1 x2 y2

  

   

    

     

    
    

     

      if t_h>t_w:
     
    
   

    

     

    
    

     

       width = t_h
     
    
   

    

     

    
    

     

       x_min=max(0,(boxes[0]+boxes[2])*0.5-width*0.5)
     
    
   

    

     

    
    

     

       x_max=min(frame.shape[1],(boxes[0]+boxes[2])*0.5+width*0.5)
     
    
   

    

     

    
    

     

       boxes[0],boxes[2]=x_min,x_max
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      else:
     
    
   

    

     

    
    

     

       width = t_w
     
    
   

    

     

    
    

     

       y_min = max(0, (boxes[1] + boxes[3]) * 0.5 - width * 0.5)
     
    
   

    

     

    
    

     

       y_max = min(frame.shape[0], (boxes[1] + boxes[3]) * 0.5 + width * 0.5)
     
    
   

    

     

    
    

     

       boxes[1], boxes[3] = y_min, y_max
     
    
  
 

图片补：

  

   

    

     

    
    

     

      img_a = cv2.imread(file)
     
    
   

    

     

    
    

     
		if img_a is None:
     
    
   

    

     

    
    

     
			continue
     
    
   

    

     

    
    

     

      		height,width=img_a.shape[:2]
     
    
   

    

     

    
    

     
		if width > height:
     
    
   

    

     

    
    

     

      			img_a = cv2.copyMakeBorder(img_a, (width - height) // 2, (width - height) // 2, 0, 0, cv2.BORDER_CONSTANT,
     
    
   

    

     

    
    

     

       value=[255, 255, 255])
     
    
   

    

     

    
    

     
		else:
     
    
   

    

     

    
    

     

      			img_a = cv2.copyMakeBorder(img_a, 0, 0, -(width - height) // 2, -(width - height) // 2, cv2.BORDER_CONSTANT,
     
    
   

    

     

    
    

     

       value=[255, 255, 255])
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      		aaaa=(max(height,width)//32+1)*32
     
    
   

    

     

    
    

     

      		img_o = cv2.resize(img_a, (aaaa, aaaa))
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      		img = cv2.cvtColor(img_o, cv2.COLOR_BGR2RGB)
     
    
  
 

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

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