Support automatic tag annotation

cvat-ai · Nov 10, 2021 · 7860d8d · 7860d8d
1 parent 7bf4154
commit 7860d8d
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/cvat/apps/lambda_manager/views.py b/cvat/apps/lambda_manager/views.py
@@ -353,6 +353,9 @@ def __init__(self, task_id):
             def append_shape(self, shape):
                 self.data["shapes"].append(shape)
 
+            def append_tag(self, tag):
+                self.data["tags"].append(tag)
+
             def submit(self):
                 if not self.is_empty():
                     serializer = LabeledDataSerializer(data=self.data)
@@ -380,7 +383,17 @@ def reset(self):
 
             for anno in annotations:
                 label_id = labels.get(anno["label"])
-                if label_id is not None:
+                if label_id is None:
+                    continue # Invalid tag provided
+                if anno["type"].lower() == "tag":
+                    results.append_tag({
+                        "frame": frame,
+                        "label_id": label_id,
+                        "source": "auto",
+                        "attributes": [],
+                        "group": None,
+                    })
+                else:
                     results.append_shape({
                         "frame": frame,
                         "label_id": label_id,