Browse Source

add image format jpg in seg transform

FlyingQianMM 4 năm trước cách đây
mục cha
commit
f8fefbd389

+ 1 - 1
deploy/openvino/python/transforms/seg_transforms.py

@@ -71,7 +71,7 @@ class Compose(SegTransform):
                 raise Exception('Can not open', img_path)
             im_data = dataset.ReadAsArray()
             return im_data.transpose((1, 2, 0))
-        elif img_format in ['jpeg', 'bmp', 'png']:
+        elif img_format in ['jpeg', 'bmp', 'png', 'jpg']:
             return cv2.imread(img_path)
         elif ext == '.npy':
             return np.load(img_path)

+ 1 - 1
paddlex/cv/transforms/seg_transforms.py

@@ -86,7 +86,7 @@ class Compose(SegTransform):
                 raise Exception('Can not open', img_path)
             im_data = dataset.ReadAsArray()
             return im_data.transpose((1, 2, 0))
-        elif img_format in ['jpeg', 'bmp', 'png']:
+        elif img_format in ['jpeg', 'bmp', 'png', 'jpg']:
             if input_channel == 3:
                 return cv2.imread(img_path, cv2.IMREAD_ANYDEPTH |
                                   cv2.IMREAD_ANYCOLOR)