-
Notifications
You must be signed in to change notification settings - Fork 328
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add VectorizedBaseImageAugmentation layer #1373
Conversation
/gcbrun |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
keras_cv/layers/preprocessing/vectorized_base_image_augmentation_layer.py
Show resolved
Hide resolved
keras_cv/layers/preprocessing/vectorized_base_image_augmentation_layer.py
Outdated
Show resolved
Hide resolved
keras_cv/layers/preprocessing/vectorized_base_image_augmentation_layer.py
Outdated
Show resolved
Hide resolved
keras_cv/layers/preprocessing/vectorized_base_image_augmentation_layer.py
Show resolved
Hide resolved
keras_cv/layers/preprocessing/vectorized_base_image_augmentation_layer.py
Show resolved
Hide resolved
keras_cv/layers/preprocessing/vectorized_base_image_augmentation_layer_test.py
Outdated
Show resolved
Hide resolved
keras_cv/layers/preprocessing/vectorized_base_image_augmentation_layer_test.py
Outdated
Show resolved
Hide resolved
/gcbrun |
1 similar comment
/gcbrun |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM once the GPU tests are passing.
keras_cv/layers/preprocessing/vectorized_base_image_augmentation_layer.py
Outdated
Show resolved
Hide resolved
/gcbrun |
/gcbrun |
Grayscale test passed --- root cause is that on GPU tf.image.grayscale is not memory efficient. in practice this barely matters because in tf.data we force CPU scope. If users request we can come up with a fix using pure tf-ops instead of tf.image ops |
* implement vectorized base image augmentation layer * Implement vectorized RandomContrast layer * KPL performance * Random contrast vectorized * Vectorized contrast * Fix vectorized base layer * Fix vectorized base layer * Add vectorized grayscale layer * Remove random contrast * Remove random contrast * test_preserves_ragged_status_Grayscale * test_preserves_ragged_status_Grayscale * Fix * Fix masks * Fix masks * rename to 'batched' * Fix docstrings * Fix docstrings * Remove ragged method * Begin ragged image support * Begin ragged image support * Begin ragged image support * Begin ragged image support * Begin ragged image support * Performance benchmark * Reformat * Vectorized grayscale * Fix ragged test case * Fix ragged test case * Fix ragged test case * Fix ragged test case * Fix ragged test case
Old non-graphmode basescale is like 100s runtime so not even comparable.