update github workflow to Publish to PyPI

39b46ea9 · 赵小蒙 · aeef64b4 · 39b46ea9
Commit 39b46ea9 authored Jun 14, 2024 by 赵小蒙
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 0 deletions

magic_pdf/model/doc_analyze_by_360layout.py magic_pdf/model/doc_analyze_by_360layout.py +8 -0

No files found.
--- a/magic_pdf/model/doc_analyze_by_360layout.py
+++ b/magic_pdf/model/doc_analyze_by_360layout.py
+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