Spaces:
Running
Running
File size: 1,679 Bytes
72049af 402b504 72049af ca46a75 72049af ca46a75 72049af 402b504 |
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 |
import csv
import numpy as np
from PIL import Image
from hivision.plugin.watermark import Watermarker, WatermarkerStyles
def csv_to_size_list(csv_file: str) -> dict:
# 初始化一个空字典
size_list_dict = {}
# 打开 CSV 文件并读取数据
with open(csv_file, mode="r", encoding="utf-8") as file:
reader = csv.reader(file)
# 跳过表头
next(reader)
# 读取数据并填充字典
for row in reader:
size_name, h, w = row
size_name_add_size = "{}\t\t({}, {})".format(size_name, h, w)
size_list_dict[size_name_add_size] = (int(h), int(w))
return size_list_dict
def csv_to_color_list(csv_file: str) -> dict:
# 初始化一个空字典
color_list_dict = {}
# 打开 CSV 文件并读取数据
with open(csv_file, mode="r", encoding="utf-8") as file:
reader = csv.reader(file)
# 跳过表头
next(reader)
# 读取数据并填充字典
for row in reader:
color_name, hex_code = row
color_list_dict[color_name] = hex_code
return color_list_dict
def range_check(value, min_value=0, max_value=255):
value = int(value)
return max(min_value, min(value, max_value))
def add_watermark(
image, text, size=50, opacity=0.5, angle=45, color="#8B8B1B", space=75
):
image = Image.fromarray(image)
watermarker = Watermarker(
input_image=image,
text=text,
style=WatermarkerStyles.STRIPED,
angle=angle,
color=color,
opacity=opacity,
size=size,
space=space,
)
return np.array(watermarker.image.convert("RGB"))
|