-
Notifications
You must be signed in to change notification settings - Fork 2k
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
[Keras-Sprint] Modification of the UNet example #1404
Conversation
Changes look good. Can you also post the latest Colab for review ? The one linked above does not seem to correspond to the PR (for ex. it has the Cosine decay LR still in it but commented out, which i do not see in the commit. Also the Colab uses PiecewiseConstantDecay which i do not see in the commit either.). |
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.
Thanks for the PR! This example is awesome -- thanks for your contributions to KerasCV to make this UX so nice 😄
""" | ||
## What does one input image and corresponding segmentation mask look like? | ||
|
||
key_rename_fn = lambda inputs: { |
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.
Is this the right name? It seems more like a rescaling / bias term operation
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.
What would you suggest?
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.
Maybe rescale_images_and_correct_masks
?
@fchollet over to you for the final review! |
I worked with Aritra on this example. This is the final version. |
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.
I worked with Aritra on this example. This is the final version.
LGTM
@martin-gorner I have also added the generated files! |
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.
Nice -- I love to see the KerasCV augmentations and visualizations 👍
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, thank you!
This PR is an effort to modify the Unet example with KerasCV.
Colab Notebook
CC: https://github.com/martin-gorner, https://github.com/margaretmz