Fixed import COCO dataset categories id

jsbroks · Feb 22, 2019 · 9cfb516 · 9cfb516
1 parent c290422
commit 9cfb516
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/app/util/task_util.py b/app/util/task_util.py
@@ -82,7 +82,7 @@ def import_coco_func(task, socket, dataset, coco_json):
             new_category.save()
 
             category_model = new_category
-            dataset.categories.append(new_category)
+            dataset.categories.append(new_category.id)
 
         task.info(f"{category_name} category found")
         # map category ids