-
Notifications
You must be signed in to change notification settings - Fork 332
Add exclude_class param to some 3D preprocessing layers. credit: lengzq #1459
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
Conversation
keras_cv/layers/preprocessing_3d/frustum_random_dropping_points.py
Outdated
Show resolved
Hide resolved
keras_cv/layers/preprocessing_3d/frustum_random_dropping_points.py
Outdated
Show resolved
Hide resolved
@@ -60,11 +60,14 @@ def __init__( | |||
theta_width, | |||
phi_width, | |||
drop_rate=None, | |||
exclude_class=-1, |
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.
nit: classes
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.
Approvod with a nit, happy to delegate on this
/gcbrun |
@ianstenbit @LukeWood just want to educate myself on this part ,why we are excluding the class params to some 3d processing layers? |
The purpose here is to allow specifying classes that shouldn't be affected by these layers. There will be an upcoming blog post in the next few months about the usage of these layers which should help demonstrate the use case! |
…zq (keras-team#1459) * Add exclude_class param to some 3D preprocessing layers. Original author: lenzq * Make exclude_class support None or a list * Rename to exclude_classes
This change has already been made downstream and I'm syncing it here (since we don't have piper -> copybara set up).