Fast image processor #28847
amyeroberts
marked this pull request as ready for review 1 year ago
qubvel
commented
on 2024-05-20
qubvel
commented
on 2024-05-22
qubvel
commented
on 2024-06-06
Draft fast image processors
68a1e757
Draft working fast version
dbf9959e
py3.8 compatible cache
3632cf70
Enable loading fast image processors through auto
49196c81
Tidy up; rescale behaviour based on input type
6fc2901e
Enable tests for fast image processors
834ae6af
Smarter rescaling
6d5c328e
Don't default to Fast
eb701c11
Safer imports
415be88d
Add necessary Pillow requirement
fb895154
Woops
d2eb99f2
Add AutoImageProcessor test
fc1530e8
Fix up
a3f6d027
Fix test for imagegpt
e0bd18df
Fix test
8c4761a8
Review comments
687da888
Add warning for TF and JAX input types
fc1e1216
Rearrange
1077938b
Return transforms
5cb11df6
NumpyToTensor transformation
fff70c3e
Rebase - include changes from upstream in ImageProcessingMixin
8b096226
Safe typing
8d82609c
Fix up
849e27bb
convert mean/std to tesnor to rescale
fdd4e5de
Don't store transforms in state
0ad7e710
Fix up
1b5885b4
Update src/transformers/image_processing_utils_fast.py
e29150ca
Update src/transformers/models/auto/image_processing_auto.py
a1f718b6
Update src/transformers/models/auto/image_processing_auto.py
af52ee2c
Update src/transformers/models/auto/image_processing_auto.py
34b8859b
Warn if fast image processor available
5e7a30d4
Update src/transformers/models/vit/image_processing_vit_fast.py
2d756071
Transpose incoming numpy images to be in CHW format
a43cabc3
Update mapping names based on packages, auto set fast to None
6acf27f4
Fix up
a38d3ee8
Fix
942286f7
Add AutoImageProcessor.from_pretrained(checkpoint, use_fast=True) test
954ee20f
Update src/transformers/models/vit/image_processing_vit_fast.py
ee06a6af
Add equivalence and speed tests
1d1d4166
Fix up
d598b5aa
amyeroberts
deleted the fast-image-processor branch 1 year ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub