From fdeb04e7bc8be5cef53c85cbfc7464b5796cb067 Mon Sep 17 00:00:00 2001 From: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com> Date: Wed, 19 Feb 2025 15:56:14 +0800 Subject: [PATCH] Fix dynamic export with YOLO World (#19304) Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com> --- ultralytics/engine/exporter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py index 242e33d9..87dd0286 100644 --- 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)