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

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

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}")
except PermissionError:
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}")

if image.mode == _SINGLE_CHANNEL_MODE:


Loading…
Cancel
Save