Verwenden Sie YOLO mit unbegrenzter Eingabe, die in eine mlpackage/mlmodel-Datei exportiert wird ⇐ Python

[Guest]

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):

Code: Select all

from ultralytics import YOLO

model = YOLO("yolo11n-seg.pt")

model.export(format="coreml")

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 .