• Vasilis Vryniotis's avatar
    Implement all AutoAugment transforms + Policies (#3123) · 83171d6a
    Vasilis Vryniotis authored
    
    
    * Invert Transform (#3104)
    
    * Adding invert operator.
    
    * Make use of the _assert_channels().
    
    * Update upper bound value.
    
    * Remove private doc from invert, create or reuse generic testing methods to avoid duplication of code in the tests. (#3106)
    
    * Create posterize transformation and refactor common methods to assist reuse. (#3108)
    
    * Implement the solarize transform. (#3112)
    
    * Implement the adjust_sharpness transform (#3114)
    
    * Adding functional operator for sharpness.
    
    * Adding transforms for sharpness.
    
    * Handling tiny images and adding a test.
    
    * Implement the autocontrast transform. (#3117)
    
    * Implement the equalize transform (#3119)
    
    * Implement the equalize transform.
    
    * Turn off deterministic for histogram.
    
    * Fixing test. (#3126)
    
    * Force ratio to be float to avoid numeric overflows on blend. (#3127)
    
    * Separate the tests of Adjust Sharpness from ColorJitter. (#3128)
    
    * Add AutoAugment Policies and main Transform (#3142)
    
    * Separate the tests of Adjust Sharpness from ColorJitter.
    
    * Initial implementation, not-jitable.
    
    * AutoAugment passing JIT.
    
    * Adding tests/docs, changing formatting.
    
    * Update test.
    
    * Fix formats
    
    * Fix documentation and imports.
    
    * Apply changes from code review:
    - Move the transformations outside of AutoAugment on a separate method.
    - Renamed degenerate method for sharpness for better clarity.
    
    * Update torchvision/transforms/functional.py
    Co-authored-by: default avatarvfdev <vfdev.5@gmail.com>
    
    * Apply more changes from code review:
    - Add InterpolationMode parameter.
    - Move all declarations away from AutoAugment constructor and into the private method.
    
    * Update documentation.
    
    * Apply suggestions from code review
    Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
    
    * Apply changes from code review:
    - Refactor code to eliminate as any to() and clamp() as possible.
    - Reuse methods where possible.
    - Apply speed ups.
    
    * Replacing pad.
    Co-authored-by: default avatarvfdev <vfdev.5@gmail.com>
    Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
    83171d6a
test_transforms_tensor.py 28 KB