基于python的图片拼接脚本

  最近想把实验截图拼接一下，奈何Mac平台上没有找到趁手的拼图工具，那不如就自己写一个吧！

#!/usr/bin/env python
# -*- coding:utf-8 -*-
"""
@author:siriyang
@file: concatImg.py
@time: 2021/12/29 10:37 
"""

import os
from PIL import Image
import argparse

def ConcatImg(input_imgs,
            direction='v',
            scale='max',
            value=0,
            border=0,
            margin=0,
            bgcolor='white',
            output_img='./out.png'):

    colors={
        "white":(255,255,255),
        "black":(0,0,0)
    }

    # 检查输入图片数量是否不少于2张
    if(len(input_imgs)<2):
        print("需要至少两张图片！")
        return
    
    # 检查输入图片路径是否合法
    for i in input_imgs:
        if not ( i.endswith(".jpg") or i.endswith(".png")):
            print(i+" 不是一张有效的图片！")
            return
    
    # 检查输出保存路径是否合法
    if not ( output_img.endswith(".jpg") or output_img.endswith(".png")):
        print(output_img+" 不是一张有效的图片保存路径！")
        return

    # 读取所有图片，并记录尺寸最大最小值
    img_list=[]
    max_w=0
    min_w=-1
    max_h=0
    min_h=-1
    for i in input_imgs:
        img=Image.open(i)
        img_list.append(img)

        if img.size[0] > max_w:
            max_w = img.size[0]

        if img.size[1] > max_h:
            max_h = img.size[1]

        if min_w==-1:
            min_w=img.size[0]
        elif img.size[0] < min_w:
            min_w=img.size[0]
        
        if min_h==-1:
            min_h=img.size[1]
        elif img.size[1] < min_h:
            min_h=img.size[1]

    res_w=0
    res_h=0

    img_resize_list=[]   
    # 缩放图片
    for img in img_list:
        if direction=='v':
            if(value>0):
                img=img.resize((value,int(value*img.size[1]/img.size[0])))
                res_w=value
            elif scale=="max":
                img=img.resize((max_w,int(max_w*img.size[1]/img.size[0])))
                res_w=max_w
            else:
                img=img.resize((min_w,int(min_w*img.size[1]/img.size[0])))
                res_w=min_w
            res_h+=img.size[1]
        else:
            if(value>0):
                img=img.resize((int(value*img.size[0]/img.size[1]),value))
                res_h=value
            elif scale=="max":
                img=img.resize((int(max_h*img.size[0]/img.size[1]),max_h))
                res_h=max_h
            else:
                img=img.resize((int(min_h*img.size[0]/img.size[1]),min_h))
                res_h=min_h
            res_w+=img.size[0]
        img_resize_list.append(img)
        
    # 加上边框宽度
    res_w+=border*2
    res_h+=border*2

    # 加上间隔宽度
    if direction=='v':
        res_h+=margin*(len(img_list)-1)
    else:
        res_w+=margin*(len(img_list)-1)
    res=Image.new('RGB', (res_w,res_h),colors[bgcolor])

    # 拼合图片
    offset=border
    for img in img_resize_list:
        if direction=='v':
            res.paste(img,(border,offset))
            offset+=img.size[1]
        else:
            res.paste(img,(offset,border))
            offset+=img.size[0]
        
        offset+=margin
    
    # 保存输出图片
    res.save(output_img)
    print("拼接完成！结果保存在 "+output_img)


if __name__ == "__main__":

    parser = argparse.ArgumentParser(description='图片拼接工具')

    parser.add_argument('imgs', nargs='+',help='要拼接的图片列表') # nargs='+' 读取至少一个参数
    parser.add_argument('-d','--direction', choices=['v','h'],default='v',help='图片拼接方向')
    group = parser.add_mutually_exclusive_group()
    group.add_argument('-s', "--scale", choices=['max','min'],default='max',help='图片拼接的拉伸方案')
    group.add_argument('-v', "--value", type=int,default=0,help='图片拼接的拉伸数值')
    parser.add_argument('-b','--border', type=int,default=0,help='外边框宽度')
    parser.add_argument('-m','--margin', type=int,default=0,help='内间隔宽度')
    parser.add_argument('-c','--bgcolor', choices=['white','black'],default='white',help='背景填充颜色')
    parser.add_argument('-o', default='./out.png',help='输出图片路径（默认为 ./out.png）')

    args = parser.parse_args()
    
    ConcatImg(args.imgs,args.direction,args.scale,args.value,args.border,args.margin,args.bgcolor,args.o)