Jelajahi Sumber

update github workflow to Publish to PyPI

赵小蒙 1 tahun lalu
induk
melakukan
39b46ea980
1 mengubah file dengan 8 tambahan dan 0 penghapusan
  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)