Views
No views yet
main with the detector trained on ScreenParse v2, which contains 1,447,100 high-quality training screenshots, leaf-element annotations, and varied viewport resolutions. The original detector trained on ScreenParse v1 is retained on the v1 branch.main branch.1from ultralytics import YOLO
2
3model = YOLO("docling-project/ScreenParser")
4results = model.predict("screenshot.png", imgsz=1280, conf=0.10, iou=0.10)
5
6for r in results:
7 for box, cls_id, conf in zip(r.boxes.xyxy, r.boxes.cls, r.boxes.conf):
8 x1, y1, x2, y2 = box.tolist()
9 label = model.names[int(cls_id)]
10 print(f"{label:20s} conf={conf:.2f} bbox=({int(x1)}, {int(y1)}, {int(x2-x1)}, {int(y2-y1)})")1import os
2from ultralytics import YOLO
3
4model = YOLO("docling-project/ScreenParser")
5image_dir = "screenshots/"
6images = sorted(
7 os.path.join(image_dir, f)
8 for f in os.listdir(image_dir)
9 if f.lower().endswith((".png", ".jpg", ".jpeg"))
10)
11
12results = model.predict(images, imgsz=1280, conf=0.10, iou=0.10, batch=16)1from ultralytics import YOLO
2
3model = YOLO("docling-project/ScreenParser")
4results = model.predict("screenshot.png", imgsz=1280, conf=0.10, iou=0.10, save=True)
5# Annotated image saved under runs/detect/predict/main checkpoint was trained on ScreenParse v2, which provides 1,447,100 high-quality training screenshots and 25,575,213 UI element annotations. The dataset uses filtered leaf-element annotations to reduce noisy nested boxes and includes multiple viewport resolutions.revision="v1".1@misc{gurbuz2026movingsparsegroundingcomplete,
2 title={ScreenParse: Moving Beyond Sparse Grounding with Complete Screen Parsing Supervision},
3 author={A. Said Gurbuz and Sunghwan Hong and Ahmed Nassar and Marc Pollefeys and Peter Staar},
4 year={2026},
5 eprint={2602.14276},
6 archivePrefix={arXiv},
7 primaryClass={cs.CV},
8 url={https://arxiv.org/abs/2602.14276},
9}