-
Notifications
You must be signed in to change notification settings - Fork 617
Connected components PR #409
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
update the forked repo
Merging master to remove testing error
WindQAQ
left a comment
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.
Hello, @sayoojbk, thanks for the contribution! Please do some changes according to comments down below. Also, it would be great to run tests locally first especially for cpp kernel part. You know... it is hard to be a human compiler haha. Thanks!
tensorflow_addons/custom_ops/image/cc/kernels/connected_components.cc
Outdated
Show resolved
Hide resolved
tensorflow_addons/custom_ops/image/cc/kernels/connected_components.h
Outdated
Show resolved
Hide resolved
tensorflow_addons/custom_ops/image/cc/kernels/connected_components.h
Outdated
Show resolved
Hide resolved
…dons into connected_components
WindQAQ
left a comment
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. I think we could rename connected_ops.py to either connected_components_ops.py or segmentation_ops.py, so does connected_ops_test.py.
Ya was about to ask you about that. In the TensorFlow contrib module this was named as |
|
As you say, could we call it from tensorflow_addons.image.connected_components import connected_componentsSo in the testing file from tensorflow_addons.image import connected_componentsand can use it directly connected_components(image) |
Ya, but when the function has to be called for example like this:
|
|
Go ahead :P Don't forget to run sanity check locally |
Ya, my bad :P Will do all the checks and let you know when it's ready for merging! |
|
Hey, @WindQAQ I guess I have made most of the changes suggested by you and ran all the tests and all have passed. It looks ready to be merged to the main branch |
WindQAQ
left a comment
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.
Great! Thanks for the contribution again!
I have ported most of the code to tf 2.0 but for some reasons the tests are not running on my system.