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

The code works well with Pytorch 1.9.0.

Related Repositories

None