solve name_set type error

- add decoded class name to the set
DDMAL · Jun 20, 2024 · abb10cd · abb10cd
1 parent 54aa0b6
commit abb10cd
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: