diff --git a/scripts/dataset_tag_editor/dte_logic.py b/scripts/dataset_tag_editor/dte_logic.py index cfae7d0..8710ce8 100644 --- a/scripts/dataset_tag_editor/dte_logic.py +++ b/scripts/dataset_tag_editor/dte_logic.py @@ -3,6 +3,8 @@ import re, sys from typing import List, Set, Optional from enum import Enum +from PIL import ImageFile +ImageFile.LOAD_TRUNCATED_IMAGES = True from PIL import Image from tqdm import tqdm