From c78354203d7bc5c211d3de30e72c0a1293a156e2 Mon Sep 17 00:00:00 2001 From: SkalskiP Date: Tue, 16 Jan 2024 17:51:56 +0100 Subject: [PATCH 1/2] ready for review --- examples/speed_estimation/requirements.txt | 8 ++++---- examples/speed_estimation/ultralytics_example.py | 4 ++++ 2 files changed, 8 insertions(+), 4 deletions(-) diff --git a/examples/speed_estimation/requirements.txt b/examples/speed_estimation/requirements.txt index 63f20432..59ca45cb 100644 --- a/examples/speed_estimation/requirements.txt +++ b/examples/speed_estimation/requirements.txt @@ -1,6 +1,6 @@ supervision==0.18.0rc1 -tqdm +tqdm==4.66.1 requests -ultralytics -super-gradients -inference +ultralytics==8.0.237 +super-gradients==3.5.0 +inference==0.9.8 diff --git a/examples/speed_estimation/ultralytics_example.py b/examples/speed_estimation/ultralytics_example.py index 6bb2b2f6..eb2c87df 100644 --- a/examples/speed_estimation/ultralytics_example.py +++ b/examples/speed_estimation/ultralytics_example.py @@ -23,12 +23,16 @@ TARGET = np.array( class ViewTransformer: + def __init__(self, source: np.ndarray, target: np.ndarray) -> None: source = source.astype(np.float32) target = target.astype(np.float32) self.m = cv2.getPerspectiveTransform(source, target) def transform_points(self, points: np.ndarray) -> np.ndarray: + if points.size == 0: + return points + reshaped_points = points.reshape(-1, 1, 2).astype(np.float32) transformed_points = cv2.perspectiveTransform(reshaped_points, self.m) return transformed_points.reshape(-1, 2) From 4d8f15f29518b7c779b07b6b2a24556756640dcb Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 16 Jan 2024 16:55:57 +0000 Subject: [PATCH 2/2] =?UTF-8?q?fix(pre=5Fcommit):=20=F0=9F=8E=A8=20auto=20?= =?UTF-8?q?format=20pre-commit=20hooks?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- examples/speed_estimation/ultralytics_example.py | 1 - 1 file changed, 1 deletion(-) diff --git a/examples/speed_estimation/ultralytics_example.py b/examples/speed_estimation/ultralytics_example.py index eb2c87df..4b0436a0 100644 --- a/examples/speed_estimation/ultralytics_example.py +++ b/examples/speed_estimation/ultralytics_example.py @@ -23,7 +23,6 @@ TARGET = np.array( class ViewTransformer: - def __init__(self, source: np.ndarray, target: np.ndarray) -> None: source = source.astype(np.float32) target = target.astype(np.float32)