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

Merge pull request #1536 from RainFrost1/cherrpy_pic

[CherryPick]fix augmix bug
......@@ -352,7 +352,7 @@ class AugMix(object):
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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册