Duplicate from Zengyf-CVer/watermarking_lab

Co-authored-by: ZengYifu <[email protected]>

.gitattributes

@@ -0,0 +1,31 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zstandard filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,64 @@
+# Gradio YOLOv5 Det
+# 创建人：曾逸夫
+# 项目地址：https://gitee.com/CV_Lab/gradio_yolov5_det
+
+# 图片格式
+*.jpg
+*.jpeg
+*.png
+*.svg
+*.gif
+
+# 视频格式
+*.mp4
+*.avi
+.ipynb_checkpoints
+/__pycache__
+*/__pycache__
+
+# 日志格式
+*.log
+*.data
+*.txt
+
+# 生成文件
+*.pdf
+*.xlsx
+*.csv
+
+# 参数文件
+*.yaml
+*.json
+
+# 压缩文件格式
+*.zip
+*.tar
+*.tar.gz
+*.rar
+
+# 字体格式
+*.ttc
+*.ttf
+*.otf
+*.pkl
+
+# 模型文件
+*.pt
+*.db
+
+/flagged
+/run
+!requirements.txt
+!cls_name/*
+!model_config/*
+!img_examples/*
+!video_examples/*
+
+!requirements.txt
+!.pre-commit-config.yaml
+
+app copy.py
+test.py
+test*.py
+
+model_download.py

README.md

@@ -0,0 +1,14 @@
+---
+title: Watermarking Lab
+emoji: 🚀
+colorFrom: pink
+colorTo: yellow
+sdk: gradio
+sdk_version: 3.1.4
+app_file: app.py
+pinned: false
+license: gpl-3.0
+duplicated_from: Zengyf-CVer/watermarking_lab
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,112 @@
+# 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()

requirements.txt

@@ -0,0 +1,3 @@
+pillow
+wget
+rich

style.css

@@ -0,0 +1,7 @@
+h1 {
+  text-align: center;
+}
+
+#content_align {
+  text-align: center;
+}

util/fonts_opt.py

@@ -0,0 +1,67 @@
+# 字体管理
+# 创建人：曾逸夫
+# 创建时间：2022-05-01
+
+import os
+import sys
+from pathlib import Path
+
+import wget
+from rich.console import Console
+
+ROOT_PATH = sys.path[0]  # 项目根目录
+
+# 中文、英文、俄语、西班牙语、阿拉伯语、韩语
+fonts_list = ["SimSun.ttf", "TimesNewRoman.ttf", "malgun.ttf"]  # 字体列表
+fonts_suffix = ["ttc", "ttf", "otf"]  # 字体后缀
+
+data_url_dict = {
+    "SimSun.ttf": "https://gitee.com/CV_Lab/gradio_yolov5_det/attach_files/1053539/download/SimSun.ttf",
+    "TimesNewRoman.ttf": "https://gitee.com/CV_Lab/gradio_yolov5_det/attach_files/1053537/download/TimesNewRoman.ttf",
+    "malgun.ttf": "https://gitee.com/CV_Lab/gradio_yolov5_det/attach_files/1053538/download/malgun.ttf",}
+
+console = Console()
+
+
+# 创建字体库
+def add_fronts(font_diff):
+
+    global font_name
+
+    for k, v in data_url_dict.items():
+        if k in font_diff:
+            font_name = v.split("/")[-1]  # 字体名称
+            Path(f"{ROOT_PATH}/fonts").mkdir(parents=True, exist_ok=True)  # 创建目录
+
+            file_path = f"{ROOT_PATH}/fonts/{font_name}"  # 字体路径
+
+            try:
+                # 下载字体文件
+                wget.download(v, file_path)
+            except Exception as e:
+                print("路径错误！程序结束！")
+                print(e)
+                sys.exit()
+            else:
+                print()
+                console.print(f"{font_name} [bold green]字体文件下载完成！[/bold green] 已保存至：{file_path}")
+
+
+# 判断字体文件
+def is_fonts(fonts_dir):
+    if os.path.isdir(fonts_dir):
+        # 如果字体库存在
+        f_list = os.listdir(fonts_dir)  # 本地字体库
+
+        font_diff = list(set(fonts_list).difference(set(f_list)))
+
+        if font_diff != []:
+            # 字体不存在
+            console.print("[bold red]字体不存在，正在加载。。。[/bold red]")
+            add_fronts(font_diff)  # 创建字体库
+        else:
+            console.print(f"{fonts_list}[bold green]字体已存在！[/bold green]")
+    else:
+        # 字体库不存在，创建字体库
+        console.print("[bold red]字体库不存在，正在创建。。。[/bold red]")
+        add_fronts(fonts_list)  # 创建字体库