Selaa lähdekoodia

Merge pull request #1260 from ucsk/develop

fix CenterCrop
will-jl944 4 vuotta sitten
vanhempi
commit
ac900869ce
1 muutettua tiedostoa jossa 1 lisäystä ja 1 poistoa
  1. 1 1
      paddlex/cv/transforms/operators.py

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

@@ -628,7 +628,7 @@ class CenterCrop(Transform):
         return image
 
     def apply_mask(self, mask):
-        mask = center_crop(mask)
+        mask = center_crop(mask, self.crop_size)
         return mask
 
     def apply(self, sample):