Merge pull request PaddlePaddle#1536 from RainFrost1/cherrpy_pic

[CherryPick]fix augmix bug
wickai · Dec 9, 2021 · 0aeca3f · 0aeca3f
2 parents 164994f + 54188b9
commit 0aeca3f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ppcls/data/preprocess/ops/operators.py b/ppcls/data/preprocess/ops/operators.py
@@ -352,7 +352,7 @@ def __call__(self, image):
  np.random.beta(self.aug_prob_coeff, self.aug_prob_coeff))
 
  # image = Image.fromarray(image)
- mix = np.zeros([image.shape[1], image.shape[0], 3])
+ mix = np.zeros(image.shape)
  for i in range(self.mixture_width):
  image_aug = image.copy()
  image_aug = Image.fromarray(image_aug)