diff --git a/ultralytics/__init__.py b/ultralytics/__init__.py index 2f8c25be..eef82497 100644 --- a/ultralytics/__init__.py +++ b/ultralytics/__init__.py @@ -1,6 +1,6 @@ # Ultralytics YOLO 🚀, AGPL-3.0 license -__version__ = "8.3.57" +__version__ = "8.3.58" import os diff --git a/ultralytics/utils/benchmarks.py b/ultralytics/utils/benchmarks.py index 2d33704a..8c79ee82 100644 --- a/ultralytics/utils/benchmarks.py +++ b/ultralytics/utils/benchmarks.py @@ -464,7 +464,7 @@ class ProfileModels: # Model and input model = YOLO(engine_file) - input_data = np.random.rand(self.imgsz, self.imgsz, 3).astype(np.float32) # must be FP32 + input_data = np.zeros((self.imgsz, self.imgsz, 3), dtype=np.uint8) # use uint8 for Classify # Warmup runs elapsed = 0.0