Browse Source

remove PIL.UnidentifiedImageError, use OSError

tags/v1.1.0
Ng Ngai Fai 5 years ago
parent
commit
d5c66f4767
1 changed files with 1 additions and 2 deletions
  1. +1
    -2
      mindinsight/explainer/encapsulator/datafile_encap.py

+ 1
- 2
mindinsight/explainer/encapsulator/datafile_encap.py View File

@@ -18,7 +18,6 @@ import os
import io import io


from PIL import Image from PIL import Image
from PIL import UnidentifiedImageError
import numpy as np import numpy as np


from mindinsight.utils.exceptions import UnknownError from mindinsight.utils.exceptions import UnknownError
@@ -91,7 +90,7 @@ class DatafileEncap(ExplainDataEncap):
raise ImageNotExistError(f"train_id:{train_id} path:{image_path} type:{image_type}") raise ImageNotExistError(f"train_id:{train_id} path:{image_path} type:{image_type}")
except PermissionError: except PermissionError:
raise FileSystemPermissionError(f"train_id:{train_id} path:{image_path} type:{image_type}") raise FileSystemPermissionError(f"train_id:{train_id} path:{image_path} type:{image_type}")
except UnidentifiedImageError:
except OSError:
raise UnknownError(f"Invalid image file: train_id:{train_id} path:{image_path} type:{image_type}") raise UnknownError(f"Invalid image file: train_id:{train_id} path:{image_path} type:{image_type}")


if image.mode == _SINGLE_CHANNEL_MODE: if image.mode == _SINGLE_CHANNEL_MODE:


Loading…
Cancel
Save