Browse Source

!932 Don't Catch PIL.UnidentifiedImageError

From: @ngtony
Reviewed-by: @wenkai_dist,@wangyue01
Signed-off-by: @wangyue01
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
80c3c4d1fc
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