未验证 提交 0aeca3f3 编写于 作者: C cuicheng01 提交者: GitHub

Merge pull request #1536 from RainFrost1/cherrpy_pic

[CherryPick]fix augmix bug
...@@ -352,7 +352,7 @@ class AugMix(object): ...@@ -352,7 +352,7 @@ class AugMix(object):
np.random.beta(self.aug_prob_coeff, self.aug_prob_coeff)) np.random.beta(self.aug_prob_coeff, self.aug_prob_coeff))
# image = Image.fromarray(image) # 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): for i in range(self.mixture_width):
image_aug = image.copy() image_aug = image.copy()
image_aug = Image.fromarray(image_aug) image_aug = Image.fromarray(image_aug)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册