-
-
Notifications
You must be signed in to change notification settings - Fork 2.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
multichannel to channel_axis (5 of 6): restoration #5288
Conversation
Otherwise, LGTM 🙂 |
Nope, I broke everything 😢 Sorry! Could you undo 4c06e3c and fix the conflicts? |
slice_at_axis is a vendored copy of numpy.lib.arraypad._slice_at_axis
If needed can revisit adding custom channel_axis support to the underlying RegionProps class
4c06e3c
to
8ff64a4
Compare
No worries, the inpainting file was modified in two other recently merged PRs as well, so that became a bit more tricky. It should be ready now if CI passes |
Thank you Greg! @jni wanna check and merge? 🙂 |
for c in range(image.shape[-1]): | ||
out[..., c] = _denoise_tv_chambolle_nd(image[..., c], weight, eps, | ||
for c in range(image.shape[channel_axis]): | ||
out[_at(c)] = _denoise_tv_chambolle_nd(image[_at(c)], weight, eps, |
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.
cute! 😂
follows #5287. Implements
channel_axis
for the restoration moduleDescription
Checklist
./doc/examples
(new features only)./benchmarks
, if your changes aren't covered by anexisting benchmark
For reviewers
later.
__init__.py
.doc/release/release_dev.rst
.