Resizer-Block_pytorch

• Implementation of Learning to Resize Images for Computer Vision Tasks.

Statement

• You can find the code of Resizer Block in lib/resizer_block.py.

• The code is tested on MNIST dataset.

• If there is something wrong in my code, please contact me, thanks!

Environment

• python 3.8.11
• pytorch 1.9.0 (other version works too, I think)

Running Steps

    python demo_MNIST_train.py

Update Records

1. The code works well with Pytorch 1.9.0.

Related Repositories

1. None