Skip to content

GitLab

Commit 4c39bcd3 authored by 赵小蒙's avatar 赵小蒙
Browse files

feat(magic-pdf): add conditional application of formula detection and recognition

parent bba53839
Show whitespace changes
Inline Side-by-side
Showing with 29 additions and 28 deletions
magic_pdf/model/pdf_extract_kit.py
View file @ 4c39bcd3
...@@ -141,6 +141,7 @@ class CustomPEKModel: ...@@ -141,6 +141,7 @@ class CustomPEKModel:
layout_cost = round(time.time() - layout_start, 2) layout_cost = round(time.time() - layout_start, 2)
logger.info(f"layout detection cost: {layout_cost}") logger.info(f"layout detection cost: {layout_cost}")
if self.apply_formula:
# 公式检测 # 公式检测
mfd_res = self.mfd_model.predict(image, imgsz=1888, conf=0.25, iou=0.45, verbose=True)[0] mfd_res = self.mfd_model.predict(image, imgsz=1888, conf=0.25, iou=0.45, verbose=True)[0]
for xyxy, conf, cla in zip(mfd_res.boxes.xyxy.cpu(), mfd_res.boxes.conf.cpu(), mfd_res.boxes.cls.cpu()): for xyxy, conf, cla in zip(mfd_res.boxes.xyxy.cpu(), mfd_res.boxes.conf.cpu(), mfd_res.boxes.cls.cpu()):
......
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