前言
最近有网友私信我，问如何把多张图片合成一张马赛克图片的样子
说是女儿从出生到现在，所有的照片，大概有上百张，所以想使用这些照片合成一张，当做生日礼物
那我们今天就用上次爬表情包的图片来做一次马赛克图片，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())`
|
完整效果
到此这篇关于Python实现多张图片合成一张马赛克图片的文章就介绍到这了
