瀏覽代碼

add mask resize target

FlyingQianMM 4 年之前
父節點
當前提交
ec47c981c0
共有 1 個文件被更改,包括 1 次插入1 次删除
  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