Browse Source

add mask resize target

FlyingQianMM 4 years ago
parent
commit
ec47c981c0
1 changed files with 1 additions and 1 deletions
  1. 1 1
      dygraph/paddlex/cv/transforms/operators.py

+ 1 - 1
dygraph/paddlex/cv/transforms/operators.py

@@ -242,7 +242,7 @@ class Resize(Transform):
         image = cv2.resize(image, target_size, interpolation=interp)
         return image
 
-    def apply_mask(self, mask):
+    def apply_mask(self, mask, target_size):
         mask = cv2.resize(mask, target_size, interpolation=cv2.INTER_NEAREST)
         return mask