Python拼接图片

Posted 2022-12-02 小丫头い

• 最近在写一篇卷积神经网络的论文，有好多实验结果需要整理，本来是用美图秀秀进行图像的拼接，但是发现重复操作太多，而且拼接效果不好，想到用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, 0, 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, dirs, files in os.walk(path):     
    for f in files :
        imagefile.append(Image.open(path+'/'+f)) 
target = Image.new('RGB', (TARGET_WIDTH, 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的图片。
  

• 写的脚本如下：

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, 0, 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, dirs, 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, dirs, 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 = []