【python】图像马赛克拼接

【python】图像马赛克拼接

最终呈现：

图像的呈现效果不是很好，因为用于拼接的图像不是很多

视频资料：
https://www.bilibili.com/video/BV1UT4y1A75d

import cv2#图片处理
import glob#文件管理
import numpy as np
from tqdm import tqdm#实现进度条
import argparse#包管理
from itertools import product#迭代器


#图片文件
def parseArgs():
    parser = argparse.ArgumentParser('拼接马赛克图片')
    parser.add_argument('--targetpath', type=str, default='des.jpg', help='目标图像路径')
    parser.add_argument('--outputpath', type=str, default='output.jpg', help='输出图像路径')
    parser.add_argument('--sourcepath', type=str, default='image', help='源图像路径')
    parser.add_argument('--blocksize', type=int, default=15, help='马赛克块大小')
    args = parser.parse_args()

    return args

#读取所有的原图片并且计算对应的颜色平均值
def readSourceImages(sourcepath, blocksize):
    print('开始读取图像')
    sourceimages = []
    avgcolors = []
    for path in tqdm(glob.glob('{}/*.jpg'.format(sourcepath))):
        #print(path)
        image = cv2.imread(path, cv2.IMREAD_COLOR)
        try:
            if image.shape[-1] != 3:
                continue
        except:
            continue
        image = cv2.resize(image, (blocksize, blocksize))
        avgcolor = np.sum(np.sum(image, axis = 0), axis=0) / (blocksize * blocksize)
        sourceimages.append(image)
        avgcolors.append(avgcolor)
    print('结束图像读取')
    return sourceimages, np.array(avgcolors)

##遍历整张图象 选取平均值最近的原图像代替此位置
def main(args):
    targetimage = cv2.imread(args.targetpath)
    outputimage = np.zeros(targetimage.shape, np.uint8)
    sourceimages, avgcolors = readSourceImages(args.sourcepath, args.blocksize)
    print('开始制作')
    for i,j in tqdm(product(range(int(targetimage.shape[1]/args.blocksize)), range(int(targetimage.shape[0]/args.blocksize)))):
        block = targetimage[j * args.blocksize: (j + 1) * args.blocksize, i *args.blocksize: (i + 1) * args.blocksize,
                :]
        avgcolor = np.sum(np.sum(block, axis = 0), axis=0) / (args.blocksize * args.blocksize)
        distance = np.linalg.norm(avgcolor - avgcolors, axis=1)
        idx = np.argmin(distance)
        outputimage[j * args.blocksize: (j + 1) * args.blocksize, i *args.blocksize: (i + 1) * args.blocksize,:] = sourceimages[idx]
    cv2.imwrite(args.outputpath, outputimage)
    cv2.imshow('result', outputimage)
    print('制作结束')

if __name__ == '__main__':
    main(parseArgs())