• amyeroberts's avatar
    Fast image processor (#28847) · f53fe35b
    amyeroberts authored
    
    
    * Draft fast image processors
    
    * Draft working fast version
    
    * py3.8 compatible cache
    
    * Enable loading fast image processors through auto
    
    * Tidy up; rescale behaviour based on input type
    
    * Enable tests for fast image processors
    
    * Smarter rescaling
    
    * Don't default to Fast
    
    * Safer imports
    
    * Add necessary Pillow requirement
    
    * Woops
    
    * Add AutoImageProcessor test
    
    * Fix up
    
    * Fix test for imagegpt
    
    * Fix test
    
    * Review comments
    
    * Add warning for TF and JAX input types
    
    * Rearrange
    
    * Return transforms
    
    * NumpyToTensor transformation
    
    * Rebase - include changes from upstream in ImageProcessingMixin
    
    * Safe typing
    
    * Fix up
    
    * convert mean/std to tesnor to rescale
    
    * Don't store transforms in state
    
    * Fix up
    
    * Update src/transformers/image_processing_utils_fast.py
    Co-authored-by: default avatarArthur <48595927+ArthurZucker@users.noreply.github.com>
    
    * Update src/transformers/models/auto/image_processing_auto.py
    Co-authored-by: default avatarArthur <48595927+ArthurZucker@users.noreply.github.com>
    
    * Update src/transformers/models/auto/image_processing_auto.py
    Co-authored-by: default avatarArthur <48595927+ArthurZucker@users.noreply.github.com>
    
    * Update src/transformers/models/auto/image_processing_auto.py
    Co-authored-by: default avatarArthur <48595927+ArthurZucker@users.noreply.github.com>
    
    * Warn if fast image processor available
    
    * Update src/transformers/models/vit/image_processing_vit_fast.py
    
    * Transpose incoming numpy images to be in CHW format
    
    * Update mapping names based on packages, auto set fast to None
    
    * Fix up
    
    * Fix
    
    * Add AutoImageProcessor.from_pretrained(checkpoint, use_fast=True) test
    
    * Update src/transformers/models/vit/image_processing_vit_fast.py
    Co-authored-by: default avatarPavel Iakubovskii <qubvel@gmail.com>
    
    * Add equivalence and speed tests
    
    * Fix up
    
    ---------
    Co-authored-by: default avatarArthur <48595927+ArthurZucker@users.noreply.github.com>
    Co-authored-by: default avatarPavel Iakubovskii <qubvel@gmail.com>
    f53fe35b
test_image_processing_videomae.py 9.17 KB