前言
最近有網友私信我，問如何把多張圖片合成一張馬賽克圖片的樣子
說是女兒從出生到現在，所有的照片，大概有上百張，所以想使用這些照片合成一張，當做生日禮物
那我們今天就用上次爬表情包的圖片來做一次馬賽克圖片，2萬張合成一張，想想就很激動
圖片素材
4K高清原圖
網頁版主頁左側可以免費領取【代碼】【相關教程、資料】，或者對于本篇文章有疑問的同學可以私信我
開發環境
- 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/395134.html
標籤：python
上一篇：一個 30 歲的程式員無比掙扎的故事，對2021年的總結
下一篇：Python+selenium 實作趣頭條的短視頻自動上傳與發布實體演示，同時支持抖音、快手、b站、小紅書等平臺的視頻自動化同步發布