forked from fire-keeper/BlindWatermark
-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathcv2_tools.py
112 lines (92 loc) · 3.53 KB
/
cv2_tools.py
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
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
__author__ = 'Erimus'
import cv2
import numpy as np
# ═══════════════════════════════════════════════
def imshow(window_name, img_data, wait=True):
'''
打印并显示图片数据
'''
img_data = img_data.copy()
print(f'\n{window_name}\n{img_data}')
print(f'{window_name} {img_data.max() = }')
print(f'{window_name} {img_data.min() = }')
img_data[img_data < 0] = 0
img_data[img_data > 255] = 255
cv2.imshow(window_name, img_data.astype(np.uint8))
if wait:
cv2.waitKey()
class cv_text():
'''
简化cv2上的文字打印，增加对齐支持。
'''
def __init__(self, font_family=None, font_size=14):
self.font = font_family or 0
(w, h), baseline = cv2.getTextSize('Bag', self.font, 1, 1)
print(f'{w=} {h=} {baseline=}')
self.base_size = h
def put(self, canvas, text, x, y, size=14, align=('bottom', 'left')):
scale = size / self.base_size
(w, h), bl = cv2.getTextSize(text, self.font,
fontScale=scale, thickness=1)
v_dict = {'bottom': 0, 'top': h, 'middle': h / 2}
h_dict = {'left': 0, 'right': w, 'center': w / 2}
for k in align:
if k in v_dict:
y += int(v_dict[k])
elif k in h_dict:
x -= int(h_dict[k])
canvas = cv2.putText(canvas, text, (x, y), self.font,
scale, (0, 0, 0), 1, cv2.LINE_AA)
return canvas
def test_text(content='Font Family'):
# 试绘文字相关的 字体 宽高 基线
font = 0
black, blue, red = (0, 0, 0), (255, 0, 0), (0, 0, 255)
pd = 50
cvs_list = []
W = H = 0 # total
scale = 1
for font in range(8):
(w, h), bl = cv2.getTextSize(f'{content}: {font}', font, scale, 1)
ch, cw = h + pd * 2, w + pd * 2
W = max(W, cw)
H += ch
cvs = np.zeros((ch, cw, 3), np.uint8)
cvs.fill(255)
cvs = cv2.putText(cvs, f'{content}: {font}', (pd, pd + h),
font, scale, black, 1, cv2.LINE_AA)
cvs = cv2.rectangle(cvs, (1, 1), (cw - 2, ch - 2), black)
cvs = cv2.rectangle(cvs, (pd, pd), (w + pd, h + pd), blue)
cvs = cv2.line(cvs, (pd, pd + h + bl), (pd + w, pd + h + bl), red)
cvs_list.append(cvs)
# cv2.imshow("cvs", cvs)
# cv2.waitKey(0)
print(f'{W = } | {H = }')
CVS = np.zeros((H, W, 3), np.uint8)
CVS.fill(255)
this_y = 0
for cvs in cvs_list:
h, w = cvs.shape[:2]
CVS[this_y:this_y + h, (W - w) // 2:(W - w) // 2 + w] = cvs
this_y += h
cv2.imshow("cvs", CVS)
cv2.waitKey(0)
def test_align():
h = w = 256
cvs = np.zeros((h * 2, w * 2, 3), np.uint8)
cvs.fill(255)
cvs = cv2.line(cvs, (0, h), (w * 2, h), (0, 0, 0))
cvs = cv2.line(cvs, (w, 0), (w, h * 2), (0, 0, 0))
cvtxt = cv_text()
size = 12
cvtxt.put(cvs, 'left top', w, h, size=size, align=('left', 'top'))
cvtxt.put(cvs, 'left bottom', w, h, size=size, align=('left', 'bottom'))
cvtxt.put(cvs, 'right top', w, h, size=size, align=('right', 'top'))
cvtxt.put(cvs, 'right bottom', w, h, size=size, align=('right', 'bottom'))
imshow('test align', cvs)
# ═══════════════════════════════════════════════
if __name__ == '__main__':
test_text()
test_align()