update run_test and run_batch

soni-aditya · Jul 6, 2021 · 1e343b0 · 1e343b0
1 parent 5b2e7a9
commit 1e343b0
Show file tree

Hide file tree

Showing 5 changed files with 2 additions and 3 deletions.
diff --git a/detect_compo/__pycache__/ip_region_proposal.cpython-35.pyc b/detect_compo/__pycache__/ip_region_proposal.cpython-35.pyc
diff --git a/detect_compo/ip_region_proposal.py b/detect_compo/ip_region_proposal.py
@@ -72,7 +72,7 @@ def compo_detection(input_img_path, output_root, uied_params,
 
     # *** Step 1 *** pre-processing: read img -> get binary map
     org, grey = pre.read_img(input_img_path, resize_by_height)
-    binary = pre.binarization(org, grad_min=int(uied_params['min-grad']), show=show, wait_key=wai_key)
+    binary = pre.binarization(org, grad_min=int(uied_params['min-grad']))
 
     # *** Step 2 *** element detection
     det.rm_line(binary, show=show, wait_key=wai_key)
@@ -115,4 +115,3 @@ def compo_detection(input_img_path, output_root, uied_params,
     file.save_corners_json(pjoin(ip_root, name + '.json'), uicompos)
     file.save_corners_json(pjoin(output_root, 'compo.json'), uicompos)
     print("[Compo Detection Completed in %.3f s] %s" % (time.clock() - start, input_img_path))
-    if show: cv2.destroyAllWindows()
diff --git a/detect_compo/lib_ip/__pycache__/Block.cpython-35.pyc b/detect_compo/lib_ip/__pycache__/Block.cpython-35.pyc
diff --git a/detect_compo/lib_ip/__pycache__/ip_detection.cpython-35.pyc b/detect_compo/lib_ip/__pycache__/ip_detection.cpython-35.pyc
diff --git a/detect_compo/lib_ip/ip_detection.py b/detect_compo/lib_ip/ip_detection.py
@@ -239,7 +239,7 @@ def is_valid_line(line):
         binary[start_row: end_row] = 0
 
     if show:
-        cv2.imshow('no-line', binary)
+        cv2.imshow('no-line binary', binary)
         if wait_key is not None:
             cv2.waitKey(wait_key)