Преглед изворни кода

feat(demo): add visualization bbox parameter and refactor parsing process

- Add is_draw_visualization_bbox parameter to enable/disable visualization of bounding boxes
- Refactor the parsing process to improve code readability and maintainability
- Update function documentation to reflect new parameter
- Simplify test code by using a more generic variable name
myhloli пре 11 месеци
родитељ
комит
17ef5c0f69
3 измењених фајлова са 3 додато и 5 уклоњено
  1. 0 0
      demo/demo1.json
  2. 0 0
      demo/demo2.json
  3. 3 5
      demo/magic_pdf_parse_main.py

Разлика између датотеке није приказан због своје велике величине
+ 0 - 0
demo/demo1.json


Разлика између датотеке није приказан због своје велике величине
+ 0 - 0
demo/demo2.json


+ 3 - 5
demo/magic_pdf_parse_main.py

@@ -69,6 +69,7 @@ def pdf_parse_main(
     :param parse_method: 解析方法, 共 auto、ocr、txt 三种,默认 auto,如果效果不好,可以尝试 ocr
     :param model_json_path: 已经存在的模型数据文件,如果为空则使用内置模型,pdf 和 model_json 务必对应
     :param is_json_md_dump: 是否将解析后的数据写入到 .json 和 .md 文件中,默认 True,会将不同阶段的数据写入到不同的 .json 文件中(共3个.json文件),md内容会保存到 .md 文件中
+    :param is_draw_visualization_bbox: 是否绘制可视化边界框,默认 True,会生成布局框和 span 框的图像
     :param output_dir: 输出结果的目录地址,会生成一个以 pdf 文件名命名的文件夹并保存所有结果
     """
     try:
@@ -97,12 +98,9 @@ def pdf_parse_main(
             model_json = []
 
         # 执行解析步骤
-        # image_writer = DiskReaderWriter(output_image_path)
         image_writer, md_writer = FileBasedDataWriter(output_image_path), FileBasedDataWriter(output_path)
 
         # 选择解析方式
-        # jso_useful_key = {"_pdf_type": "", "model_list": model_json}
-        # pipe = UNIPipe(pdf_bytes, jso_useful_key, image_writer)
         if parse_method == 'auto':
             jso_useful_key = {'_pdf_type': '', 'model_list': model_json}
             pipe = UNIPipe(pdf_bytes, jso_useful_key, image_writer)
@@ -141,5 +139,5 @@ def pdf_parse_main(
 
 # 测试
 if __name__ == '__main__':
-    pdf_path = r'D:\project\20240617magicpdf\Magic-PDF\demo\demo1.pdf'
-    pdf_parse_main(pdf_path)
+    file_path = r'D:\project\20240617magicpdf\Magic-PDF\demo\demo1.pdf'
+    pdf_parse_main(file_path)

Неке датотеке нису приказане због велике количине промена