From abb10cdf485a4d2add8c6607bb61d58083d391e7 Mon Sep 17 00:00:00 2001 From: Hanwen Zh <70364944+homework36@users.noreply.github.com> Date: Thu, 20 Jun 2024 10:00:11 -0400 Subject: [PATCH] solve name_set type error - add decoded class name to the set --- .../jobs/interactive_classifier/interactive_classifier.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/rodan-main/code/rodan/jobs/interactive_classifier/interactive_classifier.py b/rodan-main/code/rodan/jobs/interactive_classifier/interactive_classifier.py index bf820859c..4bb4d6c1d 100644 --- a/rodan-main/code/rodan/jobs/interactive_classifier/interactive_classifier.py +++ b/rodan-main/code/rodan/jobs/interactive_classifier/interactive_classifier.py @@ -289,8 +289,8 @@ def serialize_class_names_to_json(settings): for image in database: cln=image['class_name'] if type(cln)==bytes: - cln=cln.decode() - name_set.add(image['class_name']) + cln=cln.decode() + name_set.add(cln) for name in imported_class_names: if type(name)==bytes: