diff --git a/paddlex/cv/transforms/cls_transforms.py b/paddlex/cv/transforms/cls_transforms.py index 85e37bda45dc68f900e154f86342562e70d5e6b8..e83a71db0f5e0179d9dcc613140a07b3c97c15f0 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 04be89829ccff099c8625a7ffd0d05674dc93027..36bd0d2d639308a5ff2c80a379657cffbe19126e 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 79dbc2a9c769f9bfaa1018959f95eac062d919e0..0635da99868f64f771a4a9b1dff444d3c847a4de 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)