-
Notifications
You must be signed in to change notification settings - Fork 0
/
EquBeforeDivide.py
46 lines (36 loc) · 1.2 KB
/
EquBeforeDivide.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
# -*- coding:utf-8 -*-
"""
作者:机智的枫树
日期:2022年09月26日
"""
import cv2
import Process as Pro
import os
parser = Pro.get_parser()
args = parser.parse_args()
n = args.number
k = args.keyword
s = args.scale
# 根据输入参数读图
image = cv2.imread('./QRcode/'+str(n)+'.jpg')
# 读取行列值
r, c = int(s[0]), int(s[1])
print(10 * '*' + '1st STEP: Histogram' + 10 * '*')
image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
image = cv2.equalizeHist(image)
print(10 * '*' + '2nd STEP: Divide' + 10 * '*')
dividedGrey = Pro.divide(image, r, c)
# 输出图像块,如果没有blocks目录则创建目录
print(10 * '*' + '3rd STEP: Show the blocks' + 10 * '*')
if os.path.exists('./blocks/'):
Pro.del_file('./blocks/')
else:
os.makedirs('./blocks/')
Pro.show_blocks(dividedGrey)
print(10 * '*' + '4th STEP: Process the blocks' + 10 * '*')
blocksGrey = Pro.process_blocks(dividedGrey, k, equalize=False)
print(10 * '*' + '5th STEP: Combine the blocks' + 10 * '*')
assembled = Pro.avengers_assemble(blocksGrey)
print(10 * '*' + '6th STEP: Output the final QR code' + 10 * '*')
o = args.output
cv2.imwrite('./QRcode/OUTPUT_EquBf' + str(o) + '_' + str(r) + 'x' + str(c) + '.jpg', assembled)