Fix dynamic export with YOLO World (#19304)

Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
3 weeks ago · fdeb04e7bc
parent e16593336b
commit fdeb04e7bc
1 changed files with 1 additions and 1 deletions
--- a/ultralytics/engine/exporter.py
+++ b/ultralytics/engine/exporter.py
@ -520,7 +520,7 @@ class Exporter:
            if isinstance(self.model, SegmentationModel):
                dynamic["output0"] = {0: "batch", 2: "anchors"}  # shape(1, 116, 8400)
                dynamic["output1"] = {0: "batch", 2: "mask_height", 3: "mask_width"}  # shape(1,32,160,160)
-            elif isinstance(self.model, DetectionModel):
+            elif isinstance(self.model, (DetectionModel, WorldModel)):
                dynamic["output0"] = {0: "batch", 2: "anchors"}  # shape(1, 84, 8400)
            if self.args.nms:  # only batch size is dynamic with NMS
                dynamic["output0"].pop(2)