Spaces:
Runtime error
Runtime error
File size: 4,100 Bytes
787d365 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 |
# Watermarking Lab
# 创建人：曾逸夫
# 创建时间：2022-08-10
# 功能描述：单一图片加水印 透明度 字体颜色
import math
import sys
import gradio as gr
from PIL import Image, ImageDraw, ImageFont
from util.fonts_opt import is_fonts
ROOT_PATH = sys.path[0] # 根目录
DESCRIPTION = '''# Watermarking Lab'''
def Hex_to_RGB(hex):
r = int(hex[1:3], 16)
g = int(hex[3:5], 16)
b = int(hex[5:7], 16)
return r, g, b
def watermarking(img, text, text_size, text_font, wm_location, wm_trans, wm_textColor):
# 参考：https://pillow.readthedocs.io/en/stable/reference/ImageDraw.html?highlight=text#example-draw-partial-opacity-text
img_base = img.convert("RGBA")
txt = Image.new('RGBA', img_base.size, (255, 255, 255, 0))
text_size = int(text_size)
draw = ImageDraw.Draw(txt)
font = ImageFont.truetype(f"./fonts/{text_font}.ttf", text_size)
textwidth, textheight = draw.textsize(text, font)
width, height = img_base.size # 图片尺寸
# ---------- 水印位置 ----------
if wm_location == "center":
x = width / 2 - textwidth / 2
y = height / 2 - textheight / 2
elif wm_location == "bottom right":
x = width - textwidth - text_size
y = height - textheight - text_size
elif wm_location == "bottom left":
x = text_size
y = height - textheight - text_size
elif wm_location == "top right":
x = width - textwidth - text_size
y = text_size
elif wm_location == "top left":
x = text_size
y = text_size
r, g, b = Hex_to_RGB(wm_textColor)
draw.text((x, y), text, fill=(r, g, b, math.ceil(float(wm_trans) * 255)), font=font)
img_combined = Image.alpha_composite(img_base, txt)
return img_combined
def main():
is_fonts(f"{ROOT_PATH}/fonts") # 检查字体文件
with gr.Blocks(css='style.css') as gyd:
gr.Markdown(DESCRIPTION)
with gr.Row():
with gr.Column():
with gr.Row():
input_img = gr.Image(image_mode="RGB", source="upload", type="pil", label="原始图片")
with gr.Row():
wm_location = gr.Radio(choices=["center", "bottom right", "bottom left", "top right", "top left"],
value="center",
label="位置")
with gr.Row():
wm_text = gr.Textbox(value="水印内容", label="水印内容")
with gr.Row():
wm_textFont = gr.Dropdown(choices=["SimSun", "TimesNewRoman", "malgun"], value="SimSun", label="字体")
with gr.Row():
wm_textSize = gr.Number(value=50, label="文字大小")
with gr.Row():
wm_textTrans = gr.Slider(minimum=0, maximum=1, value=1, step=0.1, label="水印透明度")
with gr.Row():
wm_textColor = gr.ColorPicker(label="水印颜色")
with gr.Row():
btn_01 = gr.Button(value='加水印', variant="primary")
with gr.Column():
with gr.Row():
output_img = gr.Image(type="pil", label="水印图片")
with gr.Row():
example_list = [[
"./img_examples/bus.jpg", "Watermarking Text", 50, "TimesNewRoman", "bottom right", "0.9", "#FFFFFF"],
["./img_examples/zidane.jpg", "水印文字", 50, "SimSun", "center", "0.5", "#2488C6"]]
gr.Examples(example_list,
[input_img, wm_text, wm_textSize, wm_textFont, wm_location, wm_textTrans, wm_textColor],
output_img,
watermarking,
cache_examples=False)
btn_01.click(fn=watermarking,
inputs=[input_img, wm_text, wm_textSize, wm_textFont, wm_location, wm_textTrans, wm_textColor],
outputs=[output_img])
gyd.launch(inbrowser=True)
if __name__ == '__main__':
main()
|