From e8a219c59eeea1e3229ce643c06cb0a06113a49f Mon Sep 17 00:00:00 2001
From: kossi <120745148+kossi-kossivi@users.noreply.github.com>
Date: Thu, 13 Mar 2025 14:11:44 +0100
Subject: [PATCH] Update dataset.py
---
summit/multiview_platform/utils/dataset.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/summit/multiview_platform/utils/dataset.py b/summit/multiview_platform/utils/dataset.py
index beaf5829..98dc7a39 100644
--- a/summit/multiview_platform/utils/dataset.py
+++ b/summit/multiview_platform/utils/dataset.py
@@ -106,9 +106,9 @@ class Dataset():
return concat_views, view_limits
def select_labels(self, selected_label_names):
- selected_labels = [self.get_label_names().index(label_name.decode())
+ selected_labels = [self.get_label_names(decode=True).index(label_name.decode())
if isinstance(label_name, bytes)
- else self.get_label_names().index(label_name)
+ else self.get_label_names(decode=True).index(label_name)
for label_name in selected_label_names]
selected_indices = np.array([index
for index, label in
--
GitLab