python實現拼接圖片

阿新 • • 發佈：2020-03-24

最近在寫一篇卷積神經網路的論文，有好多實驗結果需要整理，本來是用美圖秀秀進行影象的拼接，但是發現重複操作太多，而且拼接效果不好，想到用python寫個指令碼實現，看一個簡單的例子：

python實現拼接圖片

橫向拼接

首先我需要將同一張圖片的變形拼接為一行，程式碼如下：

import os
from PIL import Image

UNIT_SIZE = 229 # 單個影象的大小為229*229
TARGET_WIDTH = 6 * UNIT_SIZE # 拼接完後的橫向長度為6*229

path = "C:/Users/zm/Desktop/FinalResult/Other-Reconstruction/2"
images = [] # 先儲存所有的影象的名稱
for root,dirs,files in os.walk(path):  
 for f in files :
  images.append(f)
for i in range(len(images)/6): # 6個影象為一組
 imagefile = []
 j = 0
 for j in range(6):
  imagefile.append(Image.open(path+'/'+images[i*6+j])) 
 target = Image.new('RGB',(TARGET_WIDTH,UNIT_SIZE)) 
 left = 0
 right = UNIT_SIZE
 for image in imagefile:  
  target.paste(image,(left,right,UNIT_SIZE))# 將image複製到target的指定位置中
  left += UNIT_SIZE # left是左上角的橫座標，依次遞增
  right += UNIT_SIZE # right是右下的橫座標，依次遞增
  quality_value = 100 # quality來指定生成圖片的質量，範圍是0～100
  target.save(path+'/result/'+os.path.splitext(images[i*6+j])[0]+'.jpg',quality = quality_value)
 imagefile = []

縱向拼接

將每種型別的圖片拼接為一行六個的圖片後再將這些圖片縱向拼接在一起，總共3種影象，那麼有3行

import os
from PIL import Image

UNIT_SIZE = 229 # 影象的高
TARGET_WIDTH = 6 * UNIT_SIZE # 一行有6個影象，那麼是6*229那麼寬

path = "C:/Users/zm/Desktop/FinalResult/Other-Reconstruction/2/result"
imagefile = [] 
for root,files in os.walk(path):  
 for f in files :
  imagefile.append(Image.open(path+'/'+f)) 
target = Image.new('RGB',UNIT_SIZE*3)) # 最終拼接的影象的大小為(229*3) * (229*6)
left = 0
right = UNIT_SIZE
for image in imagefile:  
 target.paste(image,(0,left,TARGET_WIDTH,right))
 left += UNIT_SIZE # 從上往下拼接，左上角的縱座標遞增
 right += UNIT_SIZE　#左下角的縱座標也遞增　
 quality_value = 100
 target.save(path+'/result.jpg',quality = quality_value)

同時橫向縱向拼接

今天需要處理的圖片如下：左邊是圖片所在的資料夾，每個資料夾的圖片如→_→右邊所示，需要拼接為2*5的圖片。

python實現拼接圖片

寫的指令碼如下：

import os
from PIL import Image

UNIT_SIZE = 229 # the size of image 
def pinjie(images,num):
 target = Image.new('RGB',(UNIT_SIZE*5,UNIT_SIZE*2)) # result is 2*5
 leftone = 0
 lefttwo = 0
 rightone = UNIT_SIZE
 righttwo = UNIT_SIZE
 for i in range(len(images)):
  if(i%2==0):
   target.paste(images[i],(leftone,rightone,UNIT_SIZE))
   leftone += UNIT_SIZE #第一行左上角右移
   rightone += UNIT_SIZE #右下角右移
  else:
   target.paste(images[i],(lefttwo,UNIT_SIZE,righttwo,UNIT_SIZE*2))
   lefttwo += UNIT_SIZE #第二行左上角右移
   righttwo += UNIT_SIZE #右下角右移 
 quality_value = 100
 target.save(path+dirlist[num]+'.jpg',quality = quality_value)

path = "C:/Users/laojbdao/Desktop/FinalResult/result4/different_distribution/"
dirlist = [] # all dir name
for root,files in os.walk(path):  
 for dir in dirs :
  dirlist.append(dir)

num = 0
for dir in dirlist:
 images = [] # images in each folder
 for root,files in os.walk(path+dir): # traverse each folder
  print path+dir+''
  for file in files:
   images.append(Image.open(path+dir+'/'+file))
 pinjie(images,num)
 num +=1
 images = []

python實現拼接圖片