Fixed Pixel export annotation

Author: Vaghinak Basentsyan

src/superannotate/lib/core/usecases.py

@@ -806,7 +806,7 @@ def __init__(
     ):
         super().__init__(),
         self._project = project
-        self._folder_names = folder_names
+        self._folder_names = list(folder_names)
         self._backend_service = backend_service_provider
         self._annotation_statuses = annotation_statuses
         self._include_fuse = include_fuse
@@ -856,12 +856,12 @@ def execute(self):
             if "error" in response:
                 raise AppException(response["error"])
 
-            report_message = self._project.name
+            report_message = ""
             if self._folder_names:
-                report_message = f"[{', '.join(self._folder_names)}]"
+                report_message = f"[{', '.join(self._folder_names)}] "
             logger.info(
-                f"Prepared export {response['name']} for project "
-                f"{report_message} (project ID {self._project.uuid})."
+                f"Prepared export {response['name']} for project {self._project.name} "
+                f"{report_message}(project ID {self._project.uuid})."
             )
             self._response.data = response
 
@@ -3437,19 +3437,19 @@ def execute(self):
                 )
                 if self._project.project_type == constances.ProjectType.PIXEL.value:
                     mask_path = None
-                    if os.path.exists(self._annotation_path) and not self._mask:
-                        mask_path = self._annotation_path
+                    png_path = self._annotation_path.replace("___pixel.json", "___save.png")
+                    if os.path.exists(png_path) and not self._mask:
+                        mask_path = png_path
                     elif self._mask:
                         mask_path = self._mask
 
                     if mask_path:
                         with open(mask_path, "rb") as descriptor:
-                            file = io.BytesIO(descriptor.read())
                             bucket.put_object(
                                 Key=response.data.images[image_data["id"]][
                                     "annotation_bluemap_path"
                                 ],
-                                Body=file,
+                                Body=descriptor.read(),
                             )
                 if self._verbose:
                     logger.info(

tests/integration/test_interface.py

@@ -146,3 +146,37 @@ def test_upload_images_to_project_image_quality_in_editor(self):
             [self.EXAMPLE_IMAGE_1],
             image_quality_in_editor='random_string'
         )
+
+
+class TestPixelInterface(BaseTestCase):
+    PROJECT_NAME = "Interface test"
+    TEST_FOLDER_PATH = "sample_project_pixel"
+    PROJECT_DESCRIPTION = "desc"
+    PROJECT_TYPE = "Pixel"
+    TEST_FOLDER_NAME = "folder"
+    EXAMPLE_IMAGE_1 = "example_image_1.jpg"
+
+    @property
+    def data_set_path(self):
+        return os.path.join(dirname(dirname(__file__)), "data_set")
+
+    @property
+    def folder_path(self):
+        return os.path.join(self.data_set_path, self.TEST_FOLDER_PATH)
+
+    def test_export_annotation(self):
+        sa.upload_image_to_project(self.PROJECT_NAME, f"{self.folder_path}/{self.EXAMPLE_IMAGE_1}")
+        sa.create_annotation_classes_from_classes_json(self.PROJECT_NAME, f"{self.folder_path}/classes/classes.json")
+        sa.upload_image_annotations(
+            self.PROJECT_NAME, self.EXAMPLE_IMAGE_1, f"{self.folder_path}/{self.EXAMPLE_IMAGE_1}___pixel.json"
+        )
+        with tempfile.TemporaryDirectory() as export_dir:
+            result = sa.prepare_export(
+                self.PROJECT_NAME,
+            )
+            sa.download_export(self.PROJECT_NAME, result, export_dir, True)
+            with tempfile.TemporaryDirectory() as convert_path:
+                sa.export_annotation(
+                    export_dir, convert_path, "COCO", "data_set_name", "Pixel", "panoptic_segmentation"
+                )
+                pass