Merge pull request #1182 from DDMAL/fix_1179_IC_TypeError

Solve name_set type error in interactive classifier
DDMAL · Jun 20, 2024 · 954eaee · 954eaee
2 parents 54aa0b6 + abb10cd
commit 954eaee
Showing 1 changed file with 2 additions and 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
@@ -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: