Selaa lähdekoodia

注释"清除无交点线"--无需清除,会影响后面的结果

luojiehua 2 vuotta sitten
vanhempi
commit
4312a48e4f
1 muutettua tiedostoa jossa 11 lisäystä ja 11 poistoa
  1. 11 11
      otr/table_line.py

+ 11 - 11
otr/table_line.py

@@ -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()