Skip to content

ywang370/FastImageProcessing

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Fast Image Processing with Fully-Convolutional Networks

This is a Tensorflow implementation of Fast Image Processing with Fully-Convolutional Networks.

Demo Video

https://www.youtube.com/watch?v=eQyfHgLx8Dc

Setup

Requirement

Required python libraries: Tensorflow (>=1.0) + Opencv + Numpy.

Tested in Ubuntu + Intel i7 CPU + Nvidia Titan X (Pascal) with Cuda (>=8.0) and CuDNN (>=5.0). CPU mode should also work with minor changes.

Quick Start (Testing)

  1. Clone this repository.
  2. Run "CAN24_AN/demo.py". This will generate results on L0 smoothing in "CAN24_AN/L0_smoothing/MIT-Adobe_test_1080p_result".
  3. To test a different model, change the variable "task" in "demo.py"

Training

  1. To train, change "is_training" to "True".
  2. To set up a customized training procedure, change the file paths in "prepare_data()". See the commands in the code.

Extensions

  1. The single network for all operators is "combined.py" in the folder "Single_Network". Run it and its result is in "Single_Network/result_combined/video".
  2. The parameterized network is "parameterized.py" in the folder "Parameterized_Network". Run it and its result is in "Parameterized/result_parameterized/video".

Citation

If you use our code for research, please cite our paper:

Qifeng Chen, Jia Xu, and Vladlen Koltun. Fast Image Processing with Fully-Convolutional Networks. In ICCV 2017.

License

MIT License.

About

Fast Image Processing with Fully-Convolutional Networks

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%