-
Notifications
You must be signed in to change notification settings - Fork 94
/
post_tx.py
108 lines (91 loc) · 3.45 KB
/
post_tx.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
import cv2
import numpy as np
import os
import math
def line_row_gen(img_path):
img = cv2.imread( img_path )
img_temp = np.ones_like(img) *255
gray = cv2.cvtColor( img,cv2.COLOR_BGR2GRAY )
ret, binary = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY_INV)
edges = cv2.Canny( gray,50,150,apertureSize = 3 )
# cv2.imshow('edge', edges)
# cv2.imshow('binary',binary)
# cv2.waitKey(0)
minLineLength = 100
maxLineGap = 100
lines = cv2.HoughLinesP( binary,1,np.pi/180,100,minLineLength=minLineLength,maxLineGap=maxLineGap )
try:
for line in lines:
for x1,y1,x2,y2 in line:
cv2.line( img,( x1,y1 ),( x2,y2 ),( 0,255,0 ),2 )
except:
return img_temp, 0
# points = [(box[0], box[1]), (box[2],box[1]), (box[2], box[3]), (box[0], box[3])]
# cv2.fillPoly(image,[np.array(points)],(255,0,0))
# cv2.imwrite( 'E:/image/myhoughlinesp.jpg',img )
cv2.imshow( '1',img )
cv2.waitKey(0)
return img_temp, 1
def angle(v1, v2):
dx1 = v1[2] - v1[0]
dy1 = v1[3] - v1[1]
dx2 = v2[2] - v2[0]
dy2 = v2[3] - v2[1]
angle1 = math.atan2(dy1, dx1)
angle1 = int(angle1 * 180/math.pi)
angle2 = math.atan2(dy2, dx2)
angle2 = int(angle2 * 180/math.pi)
if angle1*angle2 >= 0:
included_angle = abs(angle1-angle2)
else:
included_angle = abs(angle1) + abs(angle2)
if included_angle > 180:
included_angle = 360 - included_angle
return included_angle
def line_col_gen(img_path):
AB = [0,0,100,0]
img = cv2.imread( img_path )
img_temp = np.ones_like(img) *255
gray = cv2.cvtColor( img,cv2.COLOR_BGR2GRAY )
ret, binary = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY_INV)
edges = cv2.Canny( gray,50,150,apertureSize = 3 )
# cv2.imshow('edge', edges)
# cv2.imshow('binary',binary)
# cv2.waitKey(0)
minLineLength = 100
maxLineGap = 100
lines = cv2.HoughLinesP( binary,1,np.pi/180,100,minLineLength=20,maxLineGap=50 )
try:
for line in lines:
for x1,y1,x2,y2 in line:
CD = [x1, y1, x2, y2]
angle_cross = angle(AB, CD)
if angle_cross<90+15 and angle_cross>90-15:
cv2.line( img,( x1,y1 ),( x2,y2 ),( 255,0,0 ),2 )
except:
return img_temp, 0
# points = [(box[0], box[1]), (box[2],box[1]), (box[2], box[3]), (box[0], box[3])]
# cv2.fillPoly(image,[np.array(points)],(255,0,0))
# cv2.imwrite( 'E:/image/myhoughlinesp.jpg',img )
cv2.imshow( '2',img )
cv2.waitKey(0)
return img_temp, 1
def tx_post(row_path, nrow_path, col_path, ncol_path):
row_image, is_row_exist = line_row_gen(row_path)
nrow_image, is_nrow_exist = line_row_gen(nrow_path)
col_image, is_col_exist = line_col_gen(col_path)
ncol_image, is_ncol_exist = line_col_gen(ncol_path)
if __name__ == '__main__':
img_root = r'\result\test'
col_root = r'\result\col'
row_root = r'\result\row'
ncol_root = r'\result\ncol'
nrow_root = r'\result\nrow'
img_names = os.listdir(col_root)
for img_name in img_names:
col_path = os.path.join(col_root, img_name)
ncol_path = os.path.join(ncol_root, img_name)
row_path = os.path.join(row_root, img_name)
nrow_path = os.path.join(nrow_root, img_name)
# save_path = os.path.join(save_root, img_name)
tx_post(row_path, nrow_path, col_path, ncol_path)