Accelerated SAM decoder, fixed serverless for EXIF rotated images

CHANGELOG.md

@@ -7,7 +7,6 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## \[2.5.0] - Unreleased
 ### Added
-- New option ``semi-auto`` is available as annotations source (<https://github.com/opencv/cvat/pull/6263>) 
 - \[API\] Support for Ground Truth job creation and removal (<https://github.com/opencv/cvat/pull/6204>)
 - \[API\] Task quality estimation endpoints (<https://github.com/opencv/cvat/pull/6204>)
 
@@ -24,7 +23,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - TDB
 
 ### Fixed
-- TDB
+- Running serverless models for EXIF-rotated images (<https://github.com/opencv/cvat/pull/6275/>)
 
 ### Security
 - TDB

cvat/apps/engine/media_extractors.py

@@ -31,7 +31,6 @@
 
 ORIENTATION_EXIF_TAG = 274
 
-
 class ORIENTATION(IntEnum):
     NORMAL_HORIZONTAL=1
     MIRROR_HORIZONTAL=2
@@ -42,7 +41,6 @@ class ORIENTATION(IntEnum):
     MIRROR_HORIZONTAL_90_ROTATED=7
     NORMAL_270_ROTATED=8
 
-
 def get_mime(name):
     for type_name, type_def in MEDIA_TYPES.items():
         if type_def['has_mime_type'](name):
@@ -602,10 +600,22 @@ def __iter__(self):
                         return
 
 class IChunkWriter(ABC):
+    IMAGE_EXT = 'jpeg'
+    POINT_CLOUD_EXT = 'pcd'
+
     def __init__(self, quality, dimension=DimensionType.DIM_2D):
         self._image_quality = quality
         self._dimension = dimension
 
+    def _write_pcd_file(self, image):
+        image_buf = open(image, "rb") if isinstance(image, str) else image
+        properties = ValidateDimension.get_pcd_properties(image_buf)
+        w, h = int(properties["WIDTH"]), int(properties["HEIGHT"])
+        extension = self.POINT_CLOUD_EXT
+        image_buf.seek(0, 0)
+        image_buf = io.BytesIO(image_buf.read())
+        return image_buf, extension, w, h
+
     @staticmethod
     def _compress_image(image_path, quality):
         image = image_path.to_image() if isinstance(image_path, av.VideoFrame) else Image.open(image_path)
@@ -651,19 +661,20 @@ class ZipChunkWriter(IChunkWriter):
     def save_as_chunk(self, images, chunk_path):
         with zipfile.ZipFile(chunk_path, 'x') as zip_chunk:
             for idx, (image, path, _) in enumerate(images):
-                arcname = '{:06d}{}'.format(idx, os.path.splitext(path)[1])
-                if isinstance(image, io.BytesIO):
-                    zip_chunk.writestr(arcname, image.getvalue())
+                ext = os.path.splitext(path)[1]
+                output = io.BytesIO()
+                if self._dimension == DimensionType.DIM_2D:
+                    pil_image = rotate_within_exif(Image.open(image))
+                    pil_image.save(output, format=pil_image.format if pil_image.format else ext or self.IMAGE_EXT, quality=100, subsampling=0)
                 else:
-                    zip_chunk.write(filename=image, arcname=arcname)
+                    output, ext = self._write_pcd_file(image)[0:2]
+                arcname = '{:06d}.{}'.format(idx, ext)
+                zip_chunk.writestr(arcname, output.getvalue())
         # return empty list because ZipChunkWriter write files as is
         # and does not decode it to know img size.
         return []
 
 class ZipCompressedChunkWriter(IChunkWriter):
-    IMAGE_EXT = 'jpeg'
-    POINT_CLOUD_EXT = 'pcd'
-
     def save_as_chunk(
         self, images, chunk_path, *, compress_frames: bool = True, zip_compress_level: int = 0
     ):
@@ -680,12 +691,7 @@ def save_as_chunk(
 
                     extension = self.IMAGE_EXT
                 else:
-                    image_buf = open(image, "rb") if isinstance(image, str) else image
-                    properties = ValidateDimension.get_pcd_properties(image_buf)
-                    w, h = int(properties["WIDTH"]), int(properties["HEIGHT"])
-                    extension = self.POINT_CLOUD_EXT
-                    image_buf.seek(0, 0)
-                    image_buf = io.BytesIO(image_buf.read())
+                    image_buf, extension, w, h = self._write_pcd_file(image)
                 image_sizes.append((w, h))
                 arcname = '{:06d}.{}'.format(idx, extension)
                 zip_chunk.writestr(arcname, image_buf.getvalue())

cvat/apps/engine/task.py

@@ -897,7 +897,7 @@ def update_progress(progress):
     if validate_dimension.dimension == models.DimensionType.DIM_3D:
         kwargs["dimension"] = validate_dimension.dimension
     compressed_chunk_writer = compressed_chunk_writer_class(db_data.image_quality, **kwargs)
-    original_chunk_writer = original_chunk_writer_class(original_quality)
+    original_chunk_writer = original_chunk_writer_class(original_quality, **kwargs)
 
     # calculate chunk size if it isn't specified
     if db_data.chunk_size is None: