From 00aefd795c13bb6a5364b63798c383adc15248d4 Mon Sep 17 00:00:00 2001 From: Laughing <61612323+Laughing-q@users.noreply.github.com> Date: Mon, 6 Jan 2025 00:36:21 +0800 Subject: [PATCH] `ultralytics 8.3.58` Use `uint8` type for TensorRT Profile (#18327) Signed-off-by: Glenn Jocher Co-authored-by: Glenn Jocher Co-authored-by: Ultralytics Assistant <135830346+UltralyticsAssistant@users.noreply.github.com> --- ultralytics/__init__.py | 2 +- ultralytics/utils/benchmarks.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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