Procházet zdrojové kódy

fix: extend text block handling to include reference text in draw_bbox.py

myhloli před 2 měsíci
rodič
revize
c5b2926c7b
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  1. 1 1
      mineru/utils/draw_bbox.py

+ 1 - 1
mineru/utils/draw_bbox.py

@@ -176,7 +176,7 @@ def draw_layout_bbox(pdf_info, pdf_bytes, out_path, filename):
                         codes_caption.append(bbox)
             elif block["type"] == BlockType.TITLE:
                 titles.append(bbox)
-            elif block["type"] == BlockType.TEXT:
+            elif block["type"] in [BlockType.TEXT, BlockType.REF_TEXT]:
                 texts.append(bbox)
             elif block["type"] == BlockType.INTERLINE_EQUATION:
                 interequations.append(bbox)