前言
最近有網友私信我，問如何把多張圖片合成一張馬賽克圖片的樣子
說是女兒從出生到現在，所有的照片，大概有上百張，所以想使用這些照片合成一張，當做生日禮物
那我們今天就用上次爬表情包的圖片來做一次馬賽克圖片，2萬張合成一張，想想就很激動
圖片素材
4K高清原圖
對于本篇文章有疑問的同學可以加【資料白嫖、解答交流群：910981974】
開發環境
- Python 3.6
- Pycharm
開始代碼，先匯入所需模塊
import cv2
import glob
import argparse
import numpy as np
from tqdm import tqdm # 進度條
from itertools import product # 迭代器
讀取圖片檔案
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
讀取所有源圖片并計算對應顏色平均值
def readSourceImages(sourcepath,blocksize): print('開始讀取影像')
合法影像串列
設定一個串列，存放符合要求的顏色影像
sourceimages = []
平均顏色串列
avgcolors = []
遍歷
每遍歷一次，進度條走一次
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)
主函式
def main(args): targetimage = cv2.imread(args.targetpath) outputimage = np.zeros(targetimage.shape,np.uint8) # int8 int16 int32 int64 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) 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('制作完成')
模塊呼叫執行
if __name__ == '__main__': # run main(parseArgs())
完整效果
轉載請註明出處，本文鏈接：https://www.uj5u.com/houduan/394926.html
標籤：Python