Set images annotation statuses

Author: hovnatan

docs/source/superannotate.sdk.rst

@@ -77,6 +77,7 @@ ______
 .. autofunction:: superannotate.get_image_bytes
 .. autofunction:: superannotate.download_image
 .. autofunction:: superannotate.set_image_annotation_status
+.. autofunction:: superannotate.set_images_annotation_statuses
 .. autofunction:: superannotate.get_image_annotations
 .. autofunction:: superannotate.get_image_preannotations
 .. autofunction:: superannotate.download_image_annotations

superannotate/__init__.py

@@ -54,7 +54,8 @@ def consensus(*args, **kwargs):
 )
 from .db.project_api import (
     create_folder, delete_folders, get_folder_metadata,
-    get_project_and_folder_metadata, rename_folder, search_folders
+    get_project_and_folder_metadata, rename_folder, search_folders,
+    set_images_annotation_statuses
 )
 from .db.project_images import (
     assign_images, copy_image, copy_images, delete_images, move_image,

superannotate/db/project_api.py

@@ -82,6 +82,35 @@ def get_folder_metadata(project, folder_name):
     return res
 
 
+def get_project_and_folder_metadata(project):
+    if isinstance(project, dict):
+        project = project
+        folder = None
+    elif isinstance(project, tuple):
+        if len(project) != 2:
+            raise SAIncorrectProjectArgument(project)
+        project, folder = project
+        if not isinstance(project, dict):
+            raise SAIncorrectProjectArgument(project)
+        if folder is not None and not isinstance(project, dict):
+            raise SAIncorrectProjectArgument(project)
+    elif isinstance(project, str):
+        parts = project.split('/')
+        if len(parts) == 1:
+            project_name = parts[0]
+            project = get_project_metadata_bare(project_name)
+            folder = None
+        elif len(parts) == 2:
+            project_name, folder_name = parts
+            project = get_project_metadata_bare(project_name)
+            folder = get_folder_metadata(project, folder_name)
+        else:
+            raise SAIncorrectProjectArgument(project)
+    else:
+        raise SAIncorrectProjectArgument(project)
+    return project, folder
+
+
 def search_folders(project, folder_name=None, return_metadata=False):
     if not isinstance(project, dict):
         project = get_project_metadata_bare(project)
@@ -213,30 +242,27 @@ def rename_folder(project, new_folder_name):
     )
 
 
-def get_project_and_folder_metadata(project):
-    if isinstance(project, dict):
-        project = project
-        folder = None
-    elif isinstance(project, tuple):
-        if len(project) != 2:
-            raise SAIncorrectProjectArgument(project)
-        project, folder = project
-        if not isinstance(project, dict):
-            raise SAIncorrectProjectArgument(project)
-        if folder is not None and not isinstance(project, dict):
-            raise SAIncorrectProjectArgument(project)
-    elif isinstance(project, str):
-        parts = project.split('/')
-        if len(parts) == 1:
-            project_name = parts[0]
-            project = get_project_metadata_bare(project_name)
-            folder = None
-        elif len(parts) == 2:
-            project_name, folder_name = parts
-            project = get_project_metadata_bare(project_name)
-            folder = get_folder_metadata(project, folder_name)
-        else:
-            raise SAIncorrectProjectArgument(project)
-    else:
-        raise SAIncorrectProjectArgument(project)
-    return project, folder
+def set_images_annotation_statuses(project, image_names, annotation_status):
+    NUM_TO_SEND = 500
+    project, project_folder = get_project_and_folder_metadata(project)
+    params = {"team_id": project["team_id"], "project_id": project["id"]}
+    annotation_status = common.annotation_status_str_to_int(annotation_status)
+    data = {
+        "annotation_status": annotation_status,
+        "folder_id": project_folder["id"]
+    }
+    for start_index in range(0, len(image_names), NUM_TO_SEND):
+        end_index = min(start_index + NUM_TO_SEND, len(image_names))
+        data["image_names"] = image_names[start_index:end_index]
+        response = _api.send_request(
+            req_type='PUT',
+            path=f'/image/updateAnnotationStatusBulk',
+            params=params,
+            json_req=data
+        )
+        if not response.ok:
+            raise SABaseException(
+                response.status_code,
+                "Couldn't change annotation statuses " + response.text
+            )
+    logger.info("Annotations status changed")

superannotate/db/projects.py

@@ -5,7 +5,6 @@
 import logging
 import math
 import os
-import random
 import tempfile
 import threading
 import time

tests/test_folders.py

@@ -458,4 +458,38 @@ def test_folder_export(tmpdir):
 
     assert len(list((tmpdir / "folder2").rglob("*"))) == 2
 
-    assert len(list((tmpdir).glob("*.*"))) == 4
+    assert len(list((tmpdir).glob("*.*"))) == 4
+
+
+def test_folder_image_annotation_status(tmpdir):
+    PROJECT_NAME = "test folder set annotation status"
+    tmpdir = Path(tmpdir)
+
+    projects_found = sa.search_projects(PROJECT_NAME, return_metadata=True)
+    for pr in projects_found:
+        sa.delete_project(pr)
+
+    project = sa.create_project(PROJECT_NAME, 'test', 'Vector')
+    sa.create_annotation_classes_from_classes_json(
+        project, FROM_FOLDER / "classes" / "classes.json"
+    )
+    sa.upload_images_from_folder_to_project(
+        project, FROM_FOLDER, annotation_status="InProgress"
+    )
+    sa.create_folder(project, "folder1")
+    project = PROJECT_NAME + "/folder1"
+    sa.upload_images_from_folder_to_project(
+        project, FROM_FOLDER, annotation_status="InProgress"
+    )
+
+    sa.set_images_annotation_statuses(
+        project, ["example_image_1.jpg", "example_image_2.jpg"], "QualityCheck"
+    )
+
+    for image in ["example_image_1.jpg", "example_image_2.jpg"]:
+        metadata = sa.get_image_metadata(project, image)
+        assert metadata["annotation_status"] == "QualityCheck"
+
+    for image in ["example_image_3.jpg", "example_image_3.jpg"]:
+        metadata = sa.get_image_metadata(project, image)
+        assert metadata["annotation_status"] == "InProgress"