Explorar o código

add mask resize target

FlyingQianMM %!s(int64=4) %!d(string=hai) anos
pai
achega
ec47c981c0
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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