Skip to content
Snippets Groups Projects
Unverified Commit c4a9fe53 authored by kossi's avatar kossi Committed by GitHub
Browse files

Merge pull request #1 from kossi-kossivi/kossi-kossivi-patch-1

Update dataset.py
parents 2e5930aa e8a219c5
Branches
Tags
No related merge requests found
...@@ -106,9 +106,9 @@ class Dataset(): ...@@ -106,9 +106,9 @@ class Dataset():
return concat_views, view_limits return concat_views, view_limits
def select_labels(self, selected_label_names): 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) 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] for label_name in selected_label_names]
selected_indices = np.array([index selected_indices = np.array([index
for index, label in for index, label in
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment