Fast Image Processing with Fully-Convolutional Networks

被引:194
作者
Chen, Qifeng [1 ]
Xu, Jia [1 ]
Koltun, Vladlen [1 ]
机构
[1] Intel Labs, Santa Clara, CA 95054 USA
来源
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV) | 2017年
关键词
D O I
10.1109/ICCV.2017.273
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
We present an approach to accelerating a wide variety of image processing operators. Our approach uses a fully-convolutional network that is trained on input-output pairs that demonstrate the operator's action. After training, the original operator need not be run at all. The trained network operates at full resolution and runs in constant time. We investigate the effect of network architecture on approximation accuracy, runtime, and memory footprint, and identify a specific architecture that balances these considerations. We evaluate the presented approach on ten advanced image processing operators, including multiple variational models, multiscale tone and detail manipulation, photographic style transfer, nonlocal dehazing, and nonphotorealistic stylization. All operators are approximated by the same model. Experiments demonstrate that the presented approach is significantly more accurate than prior approximation schemes. It increases approximation accuracy as measured by PSNR across the evaluated operators by 8.5 dB on the MIT-Adobe dataset (from 27.5 to 36 dB) and reduces DSSIM by a multiplicative factor of 3 compared to the most accurate prior approximation scheme, while being the fastest. We show that our models generalize across datasets and across resolutions, and investigate a number of extensions of the presented approach.
引用
收藏
页码:2516 / 2525
页数:10
相关论文
共 79 条
  • [1] Gaussian KD-Trees for Fast High-Dimensional Filtering
    Adams, Andrew
    Gelfand, Natasha
    Dolson, Jennifer
    Levoy, Marc
    [J]. ACM TRANSACTIONS ON GRAPHICS, 2009, 28 (03):
  • [2] [Anonymous], ACM T GRAPHICS
  • [3] [Anonymous], 2016, ECCV
  • [4] [Anonymous], 2016, EUR C COMP VIS
  • [5] [Anonymous], 2017, ICLR
  • [6] [Anonymous], 2017, ICCV
  • [7] [Anonymous], 2014, ECCV
  • [8] [Anonymous], IJCV
  • [9] [Anonymous], 2008, NIPS
  • [10] [Anonymous], IEEE SIGNAL PROCESSI