Fix ValueError

pull/2/head
Vinayak Mehta 2017-04-21 20:15:35 +05:30
parent fdb4b0d494
commit bef33c75b1
1 changed files with 8 additions and 8 deletions

View File

@ -110,10 +110,10 @@ class OCRLattice:
areas = [] areas = []
for area in self.table_area: for area in self.table_area:
x1, y1, x2, y2 = area.split(",") x1, y1, x2, y2 = area.split(",")
x1 = int(x1) x1 = int(float(x1))
y1 = int(y1) y1 = int(float(y1))
x2 = int(x2) x2 = int(float(x2))
y2 = int(y2) y2 = int(float(y2))
areas.append((x1, y1, abs(x2 - x1), abs(y2 - y1))) areas.append((x1, y1, abs(x2 - x1), abs(y2 - y1)))
table_bbox = find_table_joints(areas, vmask, hmask) table_bbox = find_table_joints(areas, vmask, hmask)
else: else:
@ -265,10 +265,10 @@ class OCRStream:
table_bbox = {} table_bbox = {}
for area in self.table_area: for area in self.table_area:
x1, y1, x2, y2 = area.split(",") x1, y1, x2, y2 = area.split(",")
x1 = int(x1) x1 = int(float(x1))
y1 = int(y1) y1 = int(float(y1))
x2 = int(x2) x2 = int(float(x2))
y2 = int(y2) y2 = int(float(y2))
table_bbox[(x1, y1, x2, y2)] = None table_bbox[(x1, y1, x2, y2)] = None
else: else:
table_bbox = {(0, 0, width, height): None} table_bbox = {(0, 0, width, height): None}