20 lines
601 B
Python
20 lines
601 B
Python
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 |