visualize text content

the-laughing-monkey · Jul 6, 2021 · a0c2f9e · a0c2f9e
1 parent fe161e3
commit a0c2f9e
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 1 deletion.
diff --git a/detect_merge/Element.py b/detect_merge/Element.py
@@ -91,6 +91,8 @@ def element_relation(self, element_b, bias=0):
     def visualize_element(self, img, color=(0, 255, 0), line=1, show=False):
         loc = self.put_bbox()
         cv2.rectangle(img, loc[:2], loc[2:], color, line)
+        for child in self.children:
+            child.visualize_element(img, color=(255, 0, 255), line=line)
         if show:
             cv2.imshow('element', img)
             cv2.waitKey(0)

diff --git a/detect_merge/__pycache__/Element.cpython-35.pyc b/detect_merge/__pycache__/Element.cpython-35.pyc
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
@@ -10,7 +10,7 @@
 
 
 def show_elements(org_img, eles, show=False, win_name='element', wait_key=0, shown_resize=None, line=2):
-    color_map = {'Text':(0, 0, 255), 'Compo':(0, 255, 0), 'Block':(255, 0, 0)}
+    color_map = {'Text':(0, 0, 255), 'Compo':(0, 255, 0), 'Text Content':(255, 0, 255)}
     img = org_img.copy()
     for ele in eles:
         color = color_map[ele.category]