mirror of
https://github.com/opendatalab/MinerU.git
synced 2026-09-21 12:42:22 +08:00
23 lines
840 B
Python
23 lines
840 B
Python
from magic_pdf.config.enums import SupportedPdfParseMethod
|
|
from magic_pdf.data.dataset import Dataset
|
|
from magic_pdf.pdf_parse_union_core_v2 import pdf_parse_union
|
|
|
|
|
|
def parse_pdf_by_ocr(dataset: Dataset,
|
|
model_list,
|
|
imageWriter,
|
|
start_page_id=0,
|
|
end_page_id=None,
|
|
debug_mode=False,
|
|
lang=None,
|
|
):
|
|
return pdf_parse_union(model_list,
|
|
dataset,
|
|
imageWriter,
|
|
SupportedPdfParseMethod.OCR,
|
|
start_page_id=start_page_id,
|
|
end_page_id=end_page_id,
|
|
debug_mode=debug_mode,
|
|
lang=lang,
|
|
)
|