fix: OSError:broken data stream (#1430)

cvat-ai · Apr 20, 2020 · 9344b06 · 9344b06
1 parent 70b6f0f
commit 9344b06
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -20,6 +20,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Fixed
 - Auto annotation, TF annotation and Auto segmentation apps (https://github.com/opencv/cvat/pull/1409)
+- Import works with truncated images now: "OSError:broken data stream" on corrupt images (https://github.com/opencv/cvat/pull/1430)
 
 ### Security
 -

diff --git a/cvat/apps/engine/media_extractors.py b/cvat/apps/engine/media_extractors.py
@@ -13,7 +13,11 @@
 import av.datasets
 import numpy as np
 from pyunpack import Archive
-from PIL import Image
+from PIL import Image, ImageFile
+
+# fixes: "OSError:broken data stream" when executing line 72 while loading images downloaded from the web
+# see: https://stackoverflow.com/questions/42462431/oserror-broken-data-stream-when-reading-image-file
+ImageFile.LOAD_TRUNCATED_IMAGES = True
 
 from cvat.apps.engine.mime_types import mimetypes