Briq
/
camelot-py
6
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix debug script

pull/2/head
Vinayak Mehta 2017-04-11 20:26:01 +05:30
parent 7246e1a73d
commit dd909e2b53
1 changed files with 16 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
debug/morph_transform.py
View File

@ -1,5 +1,5 @@
""" """
usage: python morph_transform.py file.png usage: python morph_transform.py file.png scale={int} invert={bool}
finds lines present in an image using opencv's morph transform. finds lines present in an image using opencv's morph transform.
""" """
@ -22,9 +22,12 @@ def timeit(func):
return timed return timed
def mt(imagename, scale=40): def mt(imagename, scale=40, invert=False):
img = cv2.imread(imagename) img = cv2.imread(imagename)
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
if invert:
threshold = cv2.adaptiveThreshold(gray, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, cv2.THRESH_BINARY, 15, -2)
else:
threshold = cv2.adaptiveThreshold(np.invert(gray), 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, cv2.THRESH_BINARY, 15, -2) threshold = cv2.adaptiveThreshold(np.invert(gray), 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, cv2.THRESH_BINARY, 15, -2)
vertical = threshold vertical = threshold
horizontal = threshold horizontal = threshold
@ -92,7 +95,15 @@ def mt(imagename, scale=40):
@timeit @timeit
def main(): def main():
t = mt(sys.argv[1]) try:
scale = int(sys.argv[2].split('=')[1])
except IndexError:
scale = 40
try:
invert = bool(sys.argv[3].split('=')[1])
except IndexError:
invert = False
t = mt(sys.argv[1], scale=scale, invert=invert)
print 'tables found: ', len(t.keys()) print 'tables found: ', len(t.keys())