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)