Browse Source

return multi-channel image of decode in seg_transforms

FlyingQianMM 5 years ago
parent
commit
66ebf5877e
1 changed files with 1 additions and 1 deletions
  1. 1 1
      paddlex/cv/transforms/seg_transforms.py

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

@@ -86,7 +86,7 @@ class Compose(SegTransform):
             if input_channel == 3:
                 return cv2.imread(img_path)
             else:
-                im = cv2.imread(im_file, cv2.IMREAD_UNCHANGED)
+                return cv2.imread(im_file, cv2.IMREAD_UNCHANGED)
         elif ext == '.npy':
             return np.load(img_path)
         else: