Two-Stream FCNs to Balance Content and Style for Style Transfer

TL;DR

This paper introduces a two-stream FCN model for style transfer that effectively balances content preservation and style rendering, producing high-quality stylized images efficiently.

Contribution

The paper presents a novel end-to-end two-stream FCN architecture with feature injections and adaptive concatenation for improved style transfer balance.

Findings

Produces more balanced stylized images than state-of-the-art methods.

Achieves faster processing speed in style transfer.

Effectively preserves semantic content while rendering styles.

Abstract

Style transfer is to render given image contents in given styles, and it has an important role in both computer vision fundamental research and industrial applications. Following the success of deep learning based approaches, this problem has been re-launched recently, but still remains a difficult task because of trade-off between preserving contents and faithful rendering of styles. Indeed, how well-balanced content and style are is crucial in evaluating the quality of stylized images. In this paper, we propose an end-to-end two-stream Fully Convolutional Networks (FCNs) aiming at balancing the contributions of the content and the style in rendered images. Our proposed network consists of the encoder and decoder parts. The encoder part utilizes a FCN for content and a FCN for style where the two FCNs have feature injections and are independently trained to preserve the semantic content and to learn the faithful style representation in each. The semantic content feature and the style representation feature are then concatenated adaptively and fed into the decoder to generate style-transferred (stylized) images. In order to train our proposed network, we employ a loss network, the pre-trained VGG-16, to compute content loss and style loss, both of which are efficiently used for the feature injection as well as the feature concatenation. Our intensive experiments show that our proposed model generates more balanced stylized images in content and style than state-of-the-art methods. Moreover, our proposed network achieves efficiency in speed.