Views
No views yet

1from ultralytics import YOLO
2model = YOLO("/path/to/moldet_v2_yolo11n_640_general.pt") # for cpu only inference: using `moldet_v2_yolo11n_640_general.onnx` for faster speed
3model.predict("path/to/image.png", save=True, imgsz=640, conf=0.5)
1from ultralytics import YOLO
2import fitz # MuPDF
3pdf = fitz.open("doc.pdf")
4model = YOLO("/path/to/moldet_v2_yolo11n_960_doc.pt") # for cpu only inference: using `moldet_v2_yolo11n_960_doc.onnx` for faster speed
5bboxes = []
6for i, p in enumerate(pdf):
7 img = f"page_{i}.png"; p.get_pixmap().save(img)
8 for r in model.predict(img, imgsz=960, conf=0.5):
9 for box in r.boxes:
10 bboxes.append({"page":img, "conf":float(box.conf), "bbox":box.xyxy[0].tolist()})Comming soon!