# 二值化图像
|
import cv2
|
|
|
def gray_img(img):
|
result = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
|
return result
|
|
|
def is_col_empty(img, col, thresh_hold=0):
|
for r in img:
|
if r[col] > thresh_hold:
|
return False
|
return True
|
|
|
def is_col_full(img, col, thresh_hold_start=64, thresh_hold_end=255):
|
rows, cols = img.shape
|
for r in img:
|
if r[col] < thresh_hold_start or r[col] > thresh_hold_end:
|
return False
|
return True
|
|
|
# 分离同花顺的代码
|
def clip_ths_code_area(img):
|
img = gray_img(img)
|
rows, cols = img.shape
|
end_col = cols - 1
|
for col in range(0, cols, 1):
|
if is_col_full(img, col, 38, 38):
|
# print("找到分隔：", col)
|
end_col = col - 1
|
break
|
# 往前找数字分隔
|
content_start = -1
|
empty_start = -1
|
empty_end = -1
|
start_index = 0
|
end_index = end_col
|
for col in range(end_col, 0, -1):
|
if not is_col_empty(img, col):
|
if content_start < 0:
|
content_start = col
|
empty_start = -1
|
empty_end = -1
|
else:
|
if empty_start < 0:
|
empty_start = col
|
empty_end = col
|
else:
|
empty_end = col
|
if empty_start - empty_end > 2 and content_start > 0:
|
start_index = col + (empty_start - empty_end + 1)
|
end_index = content_start
|
# print("代码范围：", start_index, end_index)
|
break
|
clip_img = img[0:rows, start_index:end_index]
|
# cv2.imwrite("test1.png", clip_img)
|
return clip_img
|
|
# print(clip_img.shape)
|
# ret1, p1 = cv2.threshold(src=clip_img, thresh=100, maxval=255, type=cv2.THRESH_BINARY)
|
# cv2.imwrite("D:/workspace/GP/trade_desk/test3.png", p1)
|
|
|
if __name__ == "__main__":
|
img = gray_img(cv2.imread("C:\\Users\\Administrator\\Desktop\\test.png"))
|
cv2.imwrite("C:\\Users\\Administrator\\Desktop\\test_gray.png", img)
|
