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

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