adjust parameters

soni-aditya · Jul 5, 2021 · f50d390 · f50d390
1 parent 2ff7de7
commit f50d390
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 3 deletions.
diff --git a/detect_merge/__pycache__/merge.cpython-35.pyc b/detect_merge/__pycache__/merge.cpython-35.pyc
diff --git a/detect_merge/merge.py b/detect_merge/merge.py
@@ -81,10 +81,9 @@ def refine_elements(compos, texts, intersection_bias=2, containment_ratio=0.8):
 
 
 def remove_top_bar(elements, img_height):
-    height = img_height * 0.05
     new_elements = []
     for ele in elements:
-        if ele.row_min < height and ele.height < height:
+        if ele.row_min < 10 and ele.height < img_height * 0.05:
             continue
         new_elements.append(ele)
     return new_elements

diff --git a/detect_text/__pycache__/text_detection.cpython-35.pyc b/detect_text/__pycache__/text_detection.cpython-35.pyc
diff --git a/detect_text/text_detection.py b/detect_text/text_detection.py
@@ -109,7 +109,7 @@ def text_cvt_orc_format(ocr_result):
 def text_filter_noise(texts):
     valid_texts = []
     for text in texts:
-        if len(text.content) <= 1 and text.content.lower() not in ['a', ',', '.', '!', '?', '$', '%', ':']:
+        if len(text.content) <= 1 and text.content.lower() not in ['a', ',', '.', '!', '?', '$', '%', ':', '&']:
             continue
         valid_texts.append(text)
     return valid_texts