Add pixelshuffle layers#13571
Add pixelshuffle layers#13571ThomasDelteil merged 7 commits intoapache:masterfrom kohr-h:pixelshuffle
Conversation
|
@kohr-h, thanks for the contribution! Can you add tests for your new layers? Also can you add a bit more documentation and reference the original paper, thanks! |
|
@mxnet-label-bot Update [Operator, pr-awaiting-response] |
|
Axis swapping issue is fixed now. The 2D version now does the same as the PyTorch implementation. I'll add some more documentation. |
|
Just updated the superresolution example with the new layer, seems to work fine. The old download link wasn't working any more, so I switched to a GitHub mirror of the newer version of the dataset. |
|
Ready for review now. |
|
@apeforest @zhreshold @eric-haibin-lin Can you please help in reviewing this PR? Thanks |
stu1130
left a comment
There was a problem hiding this comment.
LGTM
@sandeep-krishnamurthy could you review it? Thanks
|
@ThomasDelteil - Can you please help review this PR? |
|
LGTM. @zhreshold @Mut1nyJD for review. @kohr-h if merging this PR would fix #13548, please add "Fixes #13548" in the PR description. this will automatically close the issue when the PR is merged. |
|
@mxnet-label-bot add [Gluon] |
|
@vandanavk I added the corresponding text. |
zhreshold
left a comment
There was a problem hiding this comment.
which looks good to me, awesome job
|
@ThomasDelteil Could you please help review/merge this PR? |
|
👍 |
* Add pixelshuffle layers, closes apache#13548 * Remove fmt comments * Use explicit class in super() * Add axis swapping to pixel shuffling, add tests * Add documentation to pixel shuffle layers * Use pixelshuffle layer and fix download in superres example * Add pixelshuffle layers to API doc page
* Add pixelshuffle layers, closes apache#13548 * Remove fmt comments * Use explicit class in super() * Add axis swapping to pixel shuffling, add tests * Add documentation to pixel shuffle layers * Use pixelshuffle layer and fix download in superres example * Add pixelshuffle layers to API doc page
* Add pixelshuffle layers, closes apache#13548 * Remove fmt comments * Use explicit class in super() * Add axis swapping to pixel shuffling, add tests * Add documentation to pixel shuffle layers * Use pixelshuffle layer and fix download in superres example * Add pixelshuffle layers to API doc page
* Add pixelshuffle layers, closes apache#13548 * Remove fmt comments * Use explicit class in super() * Add axis swapping to pixel shuffling, add tests * Add documentation to pixel shuffle layers * Use pixelshuffle layer and fix download in superres example * Add pixelshuffle layers to API doc page
Description
Pixel shuffle layers, as discussed in #13548
Checklist
Essentials
Please feel free to remove inapplicable items for your PR.
Closes #13548