Add support for DeepSparse dynamic image pipelines (#850)

obss · Mar 21, 2023 · f9bd8f3 · f9bd8f3
1 parent de68008
commit f9bd8f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sahi/models/yolov5sparse.py b/sahi/models/yolov5sparse.py
@@ -28,7 +28,7 @@ def load_model(self):
         from deepsparse import Pipeline
 
         try:
-            model = Pipeline.create(task="yolo", model_path=self.model_path)
+            model = Pipeline.create(task="yolo", model_path=self.model_path, image_size=self.image_size)
             self.set_model(model)
         except Exception as e:
             raise TypeError("Could not load the model: ", e)