|
@@ -807,11 +807,11 @@ def table_postprocess(img_new, pred, prob=0.2, is_test=0):
|
|
|
mat_plot(list_line, "delete_short_lines", is_test)
|
|
|
log("delete_short_lines " + str(time.time()-start_time))
|
|
|
|
|
|
- # 清除无交点线
|
|
|
- start_time = time.time()
|
|
|
- list_line = delete_no_cross_lines(list_line)
|
|
|
- mat_plot(list_line, "delete_no_cross_lines", is_test)
|
|
|
- log("delete_no_cross_lines " + str(time.time()-start_time))
|
|
|
+ # # 清除无交点线--无需清除,会影响后面的结果
|
|
|
+ # start_time = time.time()
|
|
|
+ # list_line = delete_no_cross_lines(list_line)
|
|
|
+ # mat_plot(list_line, "delete_no_cross_lines", is_test)
|
|
|
+ # log("delete_no_cross_lines " + str(time.time()-start_time))
|
|
|
|
|
|
# 分成横竖线
|
|
|
start_time = time.time()
|
|
@@ -920,7 +920,7 @@ def table_postprocess(img_new, pred, prob=0.2, is_test=0):
|
|
|
return [-1]
|
|
|
|
|
|
|
|
|
-def table_line(img, model, size=(512, 1024), prob=0.2, is_test=0):
|
|
|
+def table_line(img, model, size=(512, 1024), prob=0.2, is_test=1):
|
|
|
log("into table_line, prob is " + str(prob))
|
|
|
sizew, sizeh = size
|
|
|
img_new = cv2.resize(img, (sizew, sizeh), interpolation=cv2.INTER_AREA)
|
|
@@ -954,11 +954,11 @@ def table_line(img, model, size=(512, 1024), prob=0.2, is_test=0):
|
|
|
mat_plot(list_line, "delete_short_lines", is_test)
|
|
|
log("delete_short_lines " + str(time.time()-start_time))
|
|
|
|
|
|
- # 清除无交点线
|
|
|
- start_time = time.time()
|
|
|
- list_line = delete_no_cross_lines(list_line)
|
|
|
- mat_plot(list_line, "delete_no_cross_lines", is_test)
|
|
|
- log("delete_no_cross_lines " + str(time.time()-start_time))
|
|
|
+ # # 清除无交点线--无需清除,会影响后面的结果
|
|
|
+ # start_time = time.time()
|
|
|
+ # list_line = delete_no_cross_lines(list_line)
|
|
|
+ # mat_plot(list_line, "delete_no_cross_lines", is_test)
|
|
|
+ # log("delete_no_cross_lines " + str(time.time()-start_time))
|
|
|
|
|
|
# 分成横竖线
|
|
|
start_time = time.time()
|