from ultralytics import YOLO
import os

# Ordner mit Bildern
image_dir = "Test"

model = YOLO("Test/best.pt")

# Schleife über alle Dateien im Ordner
for filename in os.listdir(image_dir):
    if filename.lower().endswith((".jpg", ".jpeg", ".png")):
        image_path = os.path.join(image_dir, filename)
       
        results = model(image_path, save=True)  # Speichert Bild mit BBox

        # Alternativ: Zugriff auf erkannte Objekte
        for r in results:
            print(r.boxes.xyxy)  # Koordinaten
            print(r.boxes.conf)  # Confidence
            print(r.boxes.cls)   # Klassen