From 51d46453d37ac8fc36938136ecaaf95e28b8a723 Mon Sep 17 00:00:00 2001 From: "mikel.brostrom" Date: Thu, 17 Aug 2023 18:40:24 +0200 Subject: [PATCH 1/2] fix wrong shapes --- boxmot/utils/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/boxmot/utils/__init__.py b/boxmot/utils/__init__.py index 006a56a48e..6270f8b279 100644 --- a/boxmot/utils/__init__.py +++ b/boxmot/utils/__init__.py @@ -42,7 +42,7 @@ def wrapper(*args, **kwargs): # get tracks that are both active and in the current detections relevant_classes = active_classes.union(detected_classes) - mc_dets = np.empty(shape=(0, 7)) + mc_dets = np.empty(shape=(0, 8)) for class_id in relevant_classes: modified_args[0] = np.array( dets_dict.get(int(class_id), np.empty((0, 6))) @@ -56,7 +56,7 @@ def wrapper(*args, **kwargs): logger.debug(f"Per class updates output: {mc_dets.shape}") else: mc_dets = self.update(instance, dets, im) - + print('mc_dets.shape', mc_dets.shape) return mc_dets return wrapper From 28d6a1c6a667d06e930a844f01c818b56c762c0a Mon Sep 17 00:00:00 2001 From: "mikel.brostrom" Date: Thu, 17 Aug 2023 18:40:26 +0200 Subject: [PATCH 2/2] fix wrong shapes --- boxmot/utils/__init__.py | 1 - 1 file changed, 1 deletion(-) diff --git a/boxmot/utils/__init__.py b/boxmot/utils/__init__.py index 6270f8b279..8ada8f6771 100644 --- a/boxmot/utils/__init__.py +++ b/boxmot/utils/__init__.py @@ -56,7 +56,6 @@ def wrapper(*args, **kwargs): logger.debug(f"Per class updates output: {mc_dets.shape}") else: mc_dets = self.update(instance, dets, im) - print('mc_dets.shape', mc_dets.shape) return mc_dets return wrapper