Code Monkey home page Code Monkey logo

tf-lapsrn's Introduction

TF-LapSRN

Tensorflow implementation of LapSRN algorithm described in [1]. It can now support training for 2x, 4x, and 8x scaling factor.

To run the training:

  1. Download training dataset (DIV2K [2] [3])
    bash download_trainds.sh
  2. Run the training for 4X scaling factor
    python main.py --train --scale 4
    or
    Set training images directory
    python main.py --train --scale 4 --traindir /path/to/dir

To run the test:
python3 main.py --test --scale 4
python3 main.py --test --scale 4 --testimg /path/to/image

To export file to .pb format:

  1. Run the export script
    python3 main.py --export --scale 4


References

[1] Lai, W., Huang, J., Ahuja, N. and Yang, M. (2019). Fast and Accurate Image Super-Resolution with Deep Laplacian Pyramid Networks. Available at: https://arxiv.org/abs/1710.01992
[2] Agustsson, E., Timofte, R. (2017). NTIRE 2017 Challenge on Single Image Super-Resolution: Dataset and Study. Available at: http://www.vision.ee.ethz.ch/~timofter/publications/Agustsson-CVPRW-2017.pdf
https://data.vision.ee.ethz.ch/cvl/DIV2K/

tf-lapsrn's People

Contributors

fannymonori avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar

tf-lapsrn's Issues

TF version Version

Hello. I am attempting to retrain on a custom dataset. What version of Tensorflow is this?

What I am doing wrong?

Hi, I have tried to run your code after an admittedly very short training (just 10 epochs) and I get very strage results, as you can see in the images below, where the HR image is clearly worse than the bicubic and the original ones.
Also, I don't understand why the bicubic and HR images are not enlarged by the scale 4 factor.

The images have been obtained by running: $ python main.py --test --scale 4 --testimg ./samples/butterfly.png
As I am just starting with this super resolution stuff I wonder what I am doing wrong.

(left) original image, (center) bicubic, (right) HR image.
fullimg bicubic_image HR_image

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.