diff --git a/magic_pdf/model/doc_analyze_by_360layout.py b/magic_pdf/model/doc_analyze_by_360layout.py new file mode 100644 index 00000000..b333baf5 --- /dev/null +++ b/magic_pdf/model/doc_analyze_by_360layout.py @@ -0,0 +1,8 @@ +from ultralytics import YOLO + +image_path = '' # 待预测图片路径 +model_path = '' # 权重路径 +model = YOLO(model_path) + +result = model(image_path, save=True, conf=0.5, save_crop=False, line_width=2) +print(result) \ No newline at end of file