Commit 159fb98f authored by 赵小蒙's avatar 赵小蒙

fix:img_dir abnormal

parent 959b8d82
...@@ -84,7 +84,7 @@ def do_parse( ...@@ -84,7 +84,7 @@ def do_parse(
local_image_dir, local_md_dir = prepare_env(pdf_file_name, parse_method) local_image_dir, local_md_dir = prepare_env(pdf_file_name, parse_method)
image_writer, md_writer = DiskReaderWriter(local_image_dir), DiskReaderWriter(local_md_dir) image_writer, md_writer = DiskReaderWriter(local_image_dir), DiskReaderWriter(local_md_dir)
image_dir = (os.path.basename(local_image_dir),) image_dir = str(os.path.basename(local_image_dir))
if parse_method == "auto": if parse_method == "auto":
jso_useful_key = {"_pdf_type": "", "model_list": model_list} jso_useful_key = {"_pdf_type": "", "model_list": model_list}
...@@ -110,10 +110,7 @@ def do_parse( ...@@ -110,10 +110,7 @@ def do_parse(
if f_draw_span_bbox: if f_draw_span_bbox:
draw_span_bbox(pdf_info, pdf_bytes, local_md_dir) draw_span_bbox(pdf_info, pdf_bytes, local_md_dir)
# write_to_csv(r"D:\project\20231108code-clean\linshixuqiu\pdf_dev\新模型\新建文件夹\luanma.csv", md_content = pipe.pipe_mk_markdown(image_dir, drop_mode=DropMode.NONE)
# [pdf_file_name, pipe.pdf_mid_data['not_common_character_rate'], pipe.pdf_mid_data['not_printable_rate']])
md_content = pipe.pipe_mk_markdown(str(image_dir), drop_mode=DropMode.NONE)
if f_dump_md: if f_dump_md:
"""写markdown""" """写markdown"""
md_writer.write( md_writer.write(
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment