Fix creating chunks with original quality from png images (#7899)

cvat-ai · May 16, 2024 · 0d3dde7 · 0d3dde7
1 parent 39c2063
commit 0d3dde7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cvat/apps/engine/media_extractors.py b/cvat/apps/engine/media_extractors.py
@@ -679,7 +679,8 @@ def save_as_chunk(self, images: Iterable[tuple[Image.Image|io.IOBase|str, str, s
                             else:
                                 rot_image.save(
                                     output,
-                                    format=rot_image.format if rot_image.format else self.IMAGE_EXT,
+                                    # use format from original image, https://github.com/python-pillow/Pillow/issues/5527
+                                    format=image.format if image.format else self.IMAGE_EXT,
                                     quality=100,
                                     subsampling=0
                                 )