From a672bf79dd8e22091a9fc637199552c7282968d8 Mon Sep 17 00:00:00 2001 From: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com> Date: Fri, 6 Dec 2024 05:24:53 +0800 Subject: [PATCH] Set `dynamic` to True only if imgsz is dynamic for ONNX (#17872) Co-authored-by: Glenn Jocher Co-authored-by: Ultralytics Assistant <135830346+UltralyticsAssistant@users.noreply.github.com> --- ultralytics/engine/predictor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/engine/predictor.py b/ultralytics/engine/predictor.py index c28e1895..c5250166 100644 --- a/ultralytics/engine/predictor.py +++ b/ultralytics/engine/predictor.py @@ -155,7 +155,7 @@ class BasePredictor: same_shapes = len({x.shape for x in im}) == 1 letterbox = LetterBox( self.imgsz, - auto=same_shapes and (self.model.pt or getattr(self.model, "dynamic", False)), + auto=same_shapes and (self.model.pt or (getattr(self.model, "dynamic", False) and not self.model.imx)), stride=self.model.stride, ) return [letterbox(image=x) for x in im]