一、將原圖貼上到一張正方形的背景上

def trans_square(image):
  r"""Open the image using PIL."""
  image = image.convert('RGB')
  w,h = image.size
  background = Image.new('RGB',size=(max(w,h),max(w,h)),color=(127,127,127)) # 建立背景圖，顏色值為127
  length = int(abs(w - h) // 2) # 一側需要填充的長度
  box = (length,0) if w < h else (0,length) # 貼上的位置
  background.paste(image,box)
  return background

二、切片填充的方式使用numpy建立背景，使用切片將原圖的值填充到背景中。

 def trans_square(image):
  	r"""Open the image using PIL."""
    img = image.convert('RGB')
    img = np.array(img,dtype=np.uint8) # 圖片轉numpy
    img_h,img_w,img_c = img.shape
    if img_h != img_w:
      long_side = max(img_w,img_h)
      short_side = min(img_w,img_h)
      loc = abs(img_w - img_h) // 2
      img = img.transpose((1,2)) if img_w < img_h else img # 如果高是長邊則換軸，最後再換回來
      background = np.zeros((long_side,long_side,img_c),dtype=np.uint8) # 建立正方形背景
      background[loc: loc + short_side] = img[...] # 資料填充在中間位置
      img = background.transpose((1,2)) if img_w < img_h else background
    return Image.fromarray(img,'RGB')

使用 nn.ZeroPad2d() 或者 nn.ConstantPad2d() 進行填充

def trans_square(image):
  r"""transform square.
  :return PIL image
  """
  img = transforms.ToTensor()(image)
  C,H,W = img.shape
  pad_1 = int(abs(H - W) // 2) # 一側填充長度
  pad_2 = int(abs(H - W) - pad_1) # 另一側填充長度
  img = img.unsqueeze(0) # 加軸
  if H > W:
    img = nn.ZeroPad2d((pad_1,pad_2,0))(img) # 左右填充，填充值是0
    # img = nn.ConstantPad2d((pad_1,0),127)(img) # 左右填充，填充值是127
  elif H < W:
    img = nn.ZeroPad2d((0,pad_1,pad_2))(img) # 上下填充，填充值是0
    # img = nn.ConstantPad2d((0,pad_2),127)(img) # 上下填充，填充值是127
  img = img.squeeze(0) # 減軸
  img = transforms.ToPILImage()(img)
  return img

ps：下面看下python 將圖片轉換成九宮格形式

用到的模組PIL（安裝：pip install pillow）

完整程式碼：

from PIL import Image 
import sys 
#先將 input image 填充為正方形 
def fill_image(image): 
  width,height = image.size   
  #選取長和寬中較大值作為新圖片的 
  new_image_length = width if width > height else height   
  #生成新圖片[白底] 
  new_image = Image.new(image.mode,(new_image_length,new_image_length),color='white')  #注意這個函式！ 
  #將之前的圖貼上在新圖上，居中  
  if width > height:#原圖寬大於高，則填充圖片的豎直維度 #(x,y)二元組表示貼上上圖相對下圖的起始位置,是個座標點。 
    new_image.paste(image,(0,int((new_image_length - height) / 2))) 
  else: 
    new_image.paste(image,(int((new_image_length - width) / 2),0))   
  return new_image 
def cut_image(image):
  width,height = image.size
  item_width = int(width / 3) 
  box_list = []
  # (left,upper,right,lower)
  for i in range(0,3):
    for j in range(0,3):
      #print((i*item_width,j*item_width,(i+1)*item_width,(j+1)*item_width))
      box = (j*item_width,i*item_width,(j+1)*item_width,(i+1)*item_width)
      box_list.append(box)
  image_list = [image.crop(box) for box in box_list]
  return image_list
#儲存 
def save_images(image_list): 
  index = 1  
  for image in image_list: 
    image.save(str(index) + '.png','PNG') 
    index += 1 
if __name__ == '__main__': 
  file_path = "***"#填入圖片名 
  image = Image.open(file_path)   
  #image.show() 
  image = fill_image(image) 
  image_list = cut_image(image) 
  save_images(image_list) 

原圖：

執行程式後效果圖：

到此這篇關於使用Python將圖片轉正方形的兩種方法的文章就介紹到這了,更多相關python 圖片轉正方形內容請搜尋我們以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援我們！