前言

最近有網友私信我，問如何把多張圖片合成一張馬賽克圖片的樣子

說是女兒從出生到現在，所有的照片，大概有上百張，所以想使用這些照片合成一張，當做生日禮物

那我們今天就用上次爬表情包的圖片來做一次馬賽克圖片，2萬張合成一張，想想就很激動

圖片素材

4K高清原圖

開發環境

Python 3.6

Pycharm

實現代碼

先導入所需模塊

123456

import cv2import globimport argparseimport numpy as npfrom tqdm import tqdm # 進度條from itertools import product # 迭代器

讀取圖片文件

12345678910

def parsArgs():parser = argparse.ArgumentParser(‘拼接馬賽克圖片’)parser.add_argument(‘–targetpath’, type=str, default=’examples/3.jpg’, help=’目標圖像路徑’)parser.add_argument(‘–outputpath’, type=str, default=’output.jpg’, help=’輸出圖像的路徑’)parser.add_argument(‘–sourcepath’, type=str, default=’sourceimages’, help=’用來拼接圖像的所有源圖像文件路徑’)parser.add_argument(‘–blocksize’, type=int, default=15, help=’馬賽克快的大小’)args = parser.parse_args()return args

讀取所有源圖片並計算對應顏色平均值

12	def readSourceImages(sourcepath,blocksize):print(‘開始讀取圖像’)

合法圖像列表

設置一個列表，存放符合要求的顏色圖像

1	sourceimages = []

平均顏色列表

1	avgcolors = []

遍歷

每遍歷一次，進度條走一次

123456789101112

for path in tqdm(glob.glob(“{}/*.jpg”.format(sourcepath))):image = cv2.imread(path, cv2.IMREAD_COLOR)if image.shape[-1] != 3: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)

主函數

123456789101112131415

def main(args):targetimage = cv2.imread(args.targetpath)outputimage = np.zeros(targetimage.shape,np.uint8) # int8 int16 int32 int64sourceimages,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)distances = np.linalg.norm(avgcolor – avgcolors, axis=1)idx = np.argmin(distances)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(‘製作完成’)