Update infod_sample.py

RjDuan · Oct 6, 2021 · 35ceeb0 · 35ceeb0
1 parent 896e9da
commit 35ceeb0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/infod_sample.py b/infod_sample.py
@@ -15,6 +15,8 @@
 from utils import *
 from compression import *
 from decompression import *
+from PIL import ImageFile
+ImageFile.LOAD_TRUNCATED_IMAGES = True
 
 
 class InfoDrop(Attack):