archival-index-card-detector

from ultralytics import YOLO
from PIL import Image

model = YOLO(hf_hub_download(
    repo_id="NationalLibraryOfScotland/archival-index-card-detector",
    filename="model.pt"
))
image = Image.open("scan.jpg")
results = model.predict(image)

for i, box in enumerate(results[0].boxes):
    x1, y1, x2, y2 = box.xyxy[0].tolist()
    # Add padding (10%)
    w, h = x2 - x1, y2 - y1
    pad = 0.1
    x1 = max(0, x1 - w * pad)
    y1 = max(0, y1 - h * pad)
    x2 = min(image.width, x2 + w * pad)
    y2 = min(image.height, y2 + h * pad)

    cropped = image.crop((x1, y1, x2, y2))
    cropped.save(f"card_{i}.jpg")