Verwenden Sie YOLO mit unbegrenzter Eingabe, die in eine mlpackage/mlmodel-Datei exportiert wird
Posted: 27 Dec 2024, 10:00
Ich möchte eine .mlpackage- oder .mlmodel-Datei erstellen, die ich in Xcode importieren kann, um eine Bildsegmentierung durchzuführen. Dazu möchte ich das Segmentierungspaket in YOLO verwenden, um zu prüfen, ob es meinen Anforderungen entspricht.
Das Problem besteht nun darin, dass dieses Skript eine .mlpackage-Datei erstellt, die nur Bilder mit a akzeptiert feste Größe (640x640):
Ich möchte hier etwas ändern, wahrscheinlich mit coremltools, um unbegrenzte Bereiche zu verarbeiten (ich möchte Bilder beliebiger Größe verarbeiten). Hier ist es ein wenig beschrieben: https://apple.github.io/coremltools/doc ... ded-ranges, aber ich verstehe nicht, wie ich es mit meinem Skript implementieren kann .
Das Problem besteht nun darin, dass dieses Skript eine .mlpackage-Datei erstellt, die nur Bilder mit a akzeptiert feste Größe (640x640):
Code: Select all
from ultralytics import YOLO
model = YOLO("yolo11n-seg.pt")
model.export(format="coreml")