Skip to content

Speed up median_filter2d #304

New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

#

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Jump to bottom
Merged
merged 2 commits into from
Jun 18, 2019
Merged

Speed up median_filter2d #304

merged 2 commits into from
Jun 18, 2019

Conversation

WindQAQ
Copy link
Member

@WindQAQ WindQAQ commented Jun 17, 2019

Native tf.image.extract_patches could extract multi-channeled image patches. Speed up a lot when batch_size and channels are large.

https://colab.research.google.com/drive/1UkNXxjTMHifKmHB2N5ojOHFojtXrOUIG

@WindQAQ WindQAQ requested a review from a team June 17, 2019 18:01
@WindQAQ WindQAQ added the image label Jun 17, 2019
facaiy
facaiy approved these changes Jun 18, 2019
View reviewed changes
Copy link
Member

@facaiy facaiy left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks, looks good!

@facaiy facaiy merged commit dbadc7d into tensorflow:master Jun 18, 2019
@WindQAQ WindQAQ deleted the speed-up-median-filter2d branch June 18, 2019 11:58
# for free to join this conversation on GitHub. Already have an account? # to comment
Reviewers

@facaiy facaiy facaiy approved these changes

Assignees
No one assigned
Labels
cla: yes image
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
3 participants
@WindQAQ @facaiy @googlebot