fix truncated image reading (#4937)

deploy/pptracking/python/visualize.py

@@ -17,7 +17,8 @@ from __future__ import division
 import os
 import cv2
 import numpy as np
-from PIL import Image, ImageDraw
+from PIL import Image, ImageDraw, ImageFile
+ImageFile.LOAD_TRUNCATED_IMAGES = True
 from collections import deque
 
 

deploy/python/visualize.py

@@ -18,7 +18,8 @@ from __future__ import division
 import os
 import cv2
 import numpy as np
-from PIL import Image, ImageDraw
+from PIL import Image, ImageDraw, ImageFile
+ImageFile.LOAD_TRUNCATED_IMAGES = True
 import math
 
 

ppdet/engine/trainer.py

@@ -22,7 +22,8 @@ import copy
 import time
 
 import numpy as np
-from PIL import Image, ImageOps
+from PIL import Image, ImageOps, ImageFile
+ImageFile.LOAD_TRUNCATED_IMAGES = True
 
 import paddle
 import paddle.distributed as dist