from PIL import Image,ImageDraw,ImageFont
import numpy as np
import random
import cv2
import re
?
###################################################################################
class Make_Font(object): ?#### 設(shè)置文字水印
?? ?def __init__(self,image_path,out_path,font,font_size,diaphaneity):
?? ??? ?self.image_path = image_path ?### 讀入背景圖片
?? ??? ?self.out_path = out_path ?### 輸出水印圖片
?? ??? ?self.font = font ?### 設(shè)置水印字內(nèi)容
?? ??? ?self.font_size = font_size ?## 設(shè)置字體大小
?? ??? ?self.diaphaneity = diaphaneity ?### 設(shè)置字體透明度
?
?? ??? ?suffix = self.out_path.split('.')[-1]
?? ??? ?match = re.match(r'png',suffix)
?? ??? ?if not match:
?? ??? ??? ?raise ValueError('The out put file name must be PNG file!')
?? ?def _text_xy(self,image_size):
?? ??? ?width, height = image_size
?? ??? ?x = random.randint(min(0, width), max(0, width)) ?#### 隨機取畫文字點
?? ??? ?y = random.randint(min(0, height), max(0, height))
?? ??? ?return x,y
?
?? ?def _draw_font_box(self,image_size,font_size):
?? ??? ?img_w,img_h = image_size
?? ??? ?font_w,font_h = font_size
?? ??? ?all_x = []
?? ??? ?x = 0
?? ??? ?all_x.append(x)
?? ??? ?while x < img_w:
?? ??? ??? ?x = font_w + 50 + x ?#### ?隔50 畫一次文字
?? ??? ??? ?all_x.append(x)
?
?? ??? ?all_y = []
?? ??? ?y = 0
?? ??? ?all_y.append(y)
?? ??? ?while y < img_h:
?? ??? ??? ?y = font_h + 50 + y ? #### ?隔50 畫一次文字
?? ??? ??? ?all_y.append(y)
?? ??? ?return all_x,all_y
?
?? ?def run_make_font(self):
?? ??? ?image = Image.open(self.image_path) ## (598,419)
?? ??? ?image_x,image_y = image.size[0:2] ## (598,419)
?? ??? ?text = self.font
?? ??? ?text_diap = self.diaphaneity ? #### ?設(shè)置字體透明度 ?越小越透明 （0,100）
?? ??? ?font = ImageFont.truetype('1.ttf',self.font_size) ## 設(shè)置字體和大小
?? ??? ?layer = image.convert('RGBA') ?## 轉(zhuǎn)換圖像格式：A為透明度 ? 尺寸(598, 419)
?
?? ??? ?max_size = max(image_x,image_y)
?? ??? ?text_overlayer = Image.new('RGBA',(2*max_size,2*max_size),(255,255,255,0)) ?## 生成同等大小的透明圖片
?
?? ??? ?image_draw = ImageDraw.Draw(text_overlayer) ## 畫圖
?? ??? ?text_size_x,text_size_y = image_draw.textsize(text,font = font) ?## 獲取文本大小
?? ??? ?#print(text_size_x,text_size_y) ? ### 字體大小 （250,50）
?? ??? ?x_count,y_count = self._draw_font_box(text_overlayer.size,(text_size_x,text_size_y))
?? ??? ?for i in x_count:
?? ??? ??? ?for j in y_count:
?? ??? ??? ??? ?#text_x,text_y = text_xy((image_x,image_y)) ## 設(shè)置文本位置
?? ??? ??? ??? ?image_draw.text((int(i),int(j)),text,font=font,fill=(255,255,255,text_diap)) ## 設(shè)置文本顏色和透明度
?? ??? ?text_overlayer = text_overlayer.rotate(45) ? # 設(shè)置逆時針旋轉(zhuǎn)45度
?? ??? ?####### ?設(shè)置切割點 ?##############
?? ??? ?box_x = (text_overlayer.size[0]-image_x)/2
?? ??? ?box_y = (text_overlayer.size[1]-image_y)/2
?? ??? ?box = [box_x,box_y,box_x+image_x,box_y+image_y]
?? ??? ?new_img = text_overlayer.crop(box)
?? ??? ?new_img = new_img.resize(layer.size)
?? ??? ?#text_overlayer.save('text_overlayer_after.png') ?## 生成的水印png圖片
?? ??? ?#new_img.save('new_img.png') ?## 生成的水印png圖片
?? ??? ?after = Image.alpha_composite(layer,new_img) ?## （im1,im2）將im2復(fù)合到im1上，返回一個Image對象
?? ??? ?after.save(self.out_path) ?### .png 可以直接保存RGBA格式
?
#########################################################################################
?
if __name__=="__main__":
?? ?############### 文字水印 ?########################
?? ?MK = Make_Font(image_path='./without_water/test5.jpg',out_path='test5_after.png',font = '美團(tuán)外賣',font_size = 30,diaphaneity = 90)
?? ?MK.run_make_font()
?? ?##################################################