• vfdev's avatar
    [proto] Added some transformations and fixed type hints (#6245) · 9effc4cd
    vfdev authored
    * Another attempt to add transforms
    
    * Fixed padding type hint
    
    * Fixed fill arg for pad and rotate, affine
    
    * code formatting and type hints for affine transformation
    
    * Fixed flake8
    
    * Updated tests to save and load transforms
    
    * Fixed code formatting issue
    
    * Fixed jit loading issue
    
    * Restored fill default value to None
    Updated code according to the review
    
    * Added tests for rotation, affine and zoom transforms
    
    * Put back commented code
    
    * Random erase bypass boxes and masks
    Go back with if-return/elif-return/else-return
    
    * Fixed acceptable and non-acceptable types for Cutmix/Mixup
    
    * Updated conditions for _BaseMixupCutmix
    9effc4cd
functional_tensor.py 32.6 KB