Python拼接图片 - Go语言中文社区

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

这里写图片描述

版权声明:本文来源CSDN,感谢博主原创文章,遵循 CC 4.0 by-sa 版权协议,转载请附上原文出处链接和本声明。
原文链接:https://blog.csdn.net/zm714981790/article/details/54931909
站方申明:本站部分内容来自社区用户分享,若涉及侵权,请联系站方删除。
  • 发表于 2020-03-08 16:32:49
  • 阅读 ( 815 )
  • 分类:

0 条评论

请先 登录 后评论

官方社群

GO教程

猜你喜欢