目的：批量将多张植物照片加上植物名字后拼接成多张3*3的图片。
思路：先给给张图片命名；然后代码实现加名字和拼图操作。
结果：
代码：
# 该代码用于将图片文件名批量添加到图片上
# 并将多张图片合并拼接为1张# code1：读取一个文件中所有图片的名称
# 参考：https://blog.csdn.net/hou9876543210/article/details/115544223
import os
image_path = r"./flower_photo"
if os.path.isdir(image_path):# img_infos = os.listdir(image_path)# print(img_infos)img_infos_raw = os.listdir(image_path)# print(img_infos_raw)img_infos = [x for x in img_infos_raw if x.endswith('.jpg')]img_infos = sorted(img_infos) # 排序# print(img_infos)title = [i.split('.')[0] for i in img_infos]# print(title)
# ['logo.png', 'logo的副本.png']
# ['logo', 'logo的副本']# code2:图片指定区域涂色
# 参考：https://blog.csdn.net/baldo/article/details/124114423
import cv2
for i in range(len(img_infos)):#打开图片img = cv2.imread(os.path.join('./flower_photo/',img_infos[i]))#img(指定区域) = (RGB色)# (4640, 3472, 3)h = 38 + 500 w = 3472 - 58 img[38:h, 58:w] = (255,255,255)#save figurecv2.imwrite(os.path.join('./flower_photo_add_background/',img_infos[i]), img) # code3-在图片上添加文字
# 参考：https://blog.csdn.net/qq_42257666/article/details/122135600
# Mac下查找支持的字体 https://blog.csdn.net/weixin_47728768/article/details/115054531from PIL import Image, ImageDraw, ImageFont
# import matplotlib.pyplot as pltfor i in range(len(img_infos)):#打开图片img = Image.open(os.path.join('./flower_photo_add_background/',img_infos[i]))draw = ImageDraw.Draw(img)#设置字体和大小font_path = os.path.join("assets", "Hiragino Sans GB.ttc")myfont = ImageFont.truetype(font_path, size=400, encoding='utf-8') fillcolor = "black" #设置字体颜色#CC3299粉色#读取图片的size，也就是宽度和高度 width, height = img.size#在(30,30)的位置开始写draw.text((60,60), title[i], font=myfont, fill=fillcolor)#保存图片img.save(os.path.join('./flower_photo_add_title/',img_infos[i]))#显示图片# plt.imshow(img)# plt.show()# code4-将多张图片合并为1张
# 参考：https://blog.csdn.net/qq_40502460/article/details/120176473
# opencv,cv2:https://blog.csdn.net/fuhanghang/article/details/121260534
# import cv2
import numpy as np
import math
for i in range(int(len(title)/9)):img_list = []for j in range(9):img_path = os.path.join('./flower_photo_add_title/', img_infos[i*9+j])img = cv2.imread(img_path)h = math.floor(img.shape[0]/3)w = math.floor(img.shape[1]/3)img = cv2.resize(img, (w, h))img_list.append(img)# img0~img8# np.vstack():在竖直方向上堆叠# np.hstack():在水平方向上平铺mid1 = np.hstack((img_list[0], img_list[1], img_list[2]))mid2 = np.hstack((img_list[3], img_list[4], img_list[5]))mid3 = np.hstack((img_list[6], img_list[7], img_list[8]))img_result = np.vstack((mid1, mid2, mid3))out_path = os.path.join('./flower_photo_merge/', img_infos[i*9])cv2.imwrite(out_path, img_result)
