• Vasilis Vryniotis's avatar
    Adding AugMix implementation (#5411) · 48a61df2
    Vasilis Vryniotis authored
    
    
    * Adding basic augmix implementation.
    
    * Finish the implementation.
    
    * Add tests and documentation.
    
    * Fix tests.
    
    * Simplify code.
    
    * Speed optimizations.
    
    * Per image weights instead of per batch.
    
    * Fix tests.
    
    * Update torchvision/transforms/autoaugment.py
    Co-authored-by: default avatarvfdev <vfdev.5@gmail.com>
    
    * Changing the default severity value to get by default the same strength as RandAugment.
    Co-authored-by: default avatarvfdev <vfdev.5@gmail.com>
    48a61df2
test_transforms.py 79.7 KB