瀏覽代碼

update github workflow to Publish to PyPI

赵小蒙 1 年之前
父節點
當前提交
39b46ea980
共有 1 個文件被更改,包括 8 次插入0 次删除
  1. 8 0
      magic_pdf/model/doc_analyze_by_360layout.py

+ 8 - 0
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)