瀏覽代碼

is_test=0

luojiehua 2 年之前
父節點
當前提交
f361e7d9e9
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      otr/table_line.py

+ 1 - 1
otr/table_line.py

@@ -697,7 +697,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=1):
+def table_line(img, model, size=(512, 1024), prob=0.2, is_test=0):
     log("into table_line, prob is " + str(prob))
     sizew, sizeh = size
     img_new = cv2.resize(img, (sizew, sizeh), interpolation=cv2.INTER_AREA)