Hi, I noticed that the implementation of the augment_and_mix function is different from the official code of Augmix. My guess is that the `mix += ws[i] * image_aug` part shouldn't be in the second for loop. Hope this will help.
Hi,
I noticed that the implementation of the augment_and_mix function is different from the official code of Augmix.
My guess is that the
mix += ws[i] * image_augpart shouldn't be in the second for loop.Hope this will help.