modify error message for training

paddlex/cv/transforms/cls_transforms.py

@@ -48,8 +48,9 @@ class Compose:
             tuple: 根据网络所需字段所组成的tuple；
                 字段由transforms中的最后一个数据预处理操作决定。
         """
+        try:
             im = cv2.imread(im).astype('float32')
-        if im is None:
+        except:
             raise TypeError('Can\'t read The image file {}!'.format(im))
         im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB)
         for op in self.transforms:

paddlex/cv/transforms/det_transforms.py

@@ -73,8 +73,9 @@ class Compose:
         def decode_image(im_file, im_info, label_info):
             if im_info is None:
                 im_info = dict()
+            try:
                 im = cv2.imread(im_file).astype('float32')
-            if im is None:
+            except:
                 raise TypeError(
                    'Can\'t read The image file {}!'.format(im_file))
             im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB)

paddlex/cv/transforms/seg_transforms.py

@@ -59,8 +59,9 @@ class Compose:
 
         if im_info is None:
             im_info = dict()
+        try:
             im = cv2.imread(im).astype('float32')
-        if im is None:
+        except:
             raise ValueError('Can\'t read The image file {}!'.format(im))
         if self.to_rgb:
             im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB)