소스 검색

add image format jpg in seg transform

FlyingQianMM 4 년 전
부모
커밋
f8fefbd389
2개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 1 1
      deploy/openvino/python/transforms/seg_transforms.py
  2. 1 1
      paddlex/cv/transforms/seg_transforms.py

+ 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)