From afe9588187a2da37898eebf3a705d39f18dd95b5 Mon Sep 17 00:00:00 2001
From: jiangjiajun <jiangjiajun@baidu.com>
Date: Tue, 28 Apr 2020 17:50:33 +0800
Subject: [PATCH] modify error message for training

---
 paddlex/cv/transforms/cls_transforms.py | 5 +++--
 paddlex/cv/transforms/det_transforms.py | 7 ++++---
 paddlex/cv/transforms/seg_transforms.py | 5 +++--
 3 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/paddlex/cv/transforms/cls_transforms.py b/paddlex/cv/transforms/cls_transforms.py
index 85e37bd..e83a71d 100644
--- a/paddlex/cv/transforms/cls_transforms.py
+++ b/paddlex/cv/transforms/cls_transforms.py
@@ -48,8 +48,9 @@ class Compose:
             tuple: 根据网络所需字段所组成的tuple；
                 字段由transforms中的最后一个数据预处理操作决定。
         """
-        im = cv2.imread(im).astype('float32')
-        if im is None:
+        try:
+            im = cv2.imread(im).astype('float32')
+        except:
             raise TypeError('Can\'t read The image file {}!'.format(im))
         im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB)
         for op in self.transforms:
diff --git a/paddlex/cv/transforms/det_transforms.py b/paddlex/cv/transforms/det_transforms.py
index 04be898..36bd0d2 100644
--- a/paddlex/cv/transforms/det_transforms.py
+++ b/paddlex/cv/transforms/det_transforms.py
@@ -73,10 +73,11 @@ class Compose:
         def decode_image(im_file, im_info, label_info):
             if im_info is None:
                 im_info = dict()
-            im = cv2.imread(im_file).astype('float32')
-            if im is None:
+            try:
+                im = cv2.imread(im_file).astype('float32')
+            except:
                 raise TypeError(
-                    'Can\'t read The image file {}!'.format(im_file))
+                   'Can\'t read The image file {}!'.format(im_file))
             im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB)
             # make default im_info with [h, w, 1]
             im_info['im_resize_info'] = np.array(
diff --git a/paddlex/cv/transforms/seg_transforms.py b/paddlex/cv/transforms/seg_transforms.py
index 79dbc2a..0635da9 100644
--- a/paddlex/cv/transforms/seg_transforms.py
+++ b/paddlex/cv/transforms/seg_transforms.py
@@ -59,8 +59,9 @@ class Compose:
 
         if im_info is None:
             im_info = dict()
-        im = cv2.imread(im).astype('float32')
-        if im is None:
+        try:
+            im = cv2.imread(im).astype('float32')
+        except:
             raise ValueError('Can\'t read The image file {}!'.format(im))
         if self.to_rgb:
             im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB)
-- 
GitLab