/Lightweight-Image-Style-Transfer-Model

Use all kinds of neural network pruning methods to obtain a lightweight fast image style transfer model

Primary LanguageC#MIT LicenseMIT

Lightweight-Image-Style-Transfer-Model

In this work, we design a image transfer network, and use all kinds of network prunning methods to obtain a lightweight fast image style transfer model which is suitable to released into mobile devices. Figure 1

Figure 1. The overall structure of the fast image style transfer model which consists of two compartments, a image transfer network and a features extraction network.

Figure 2

Figure 2. The proposed image transfer network without pruning.

Figure 3

Figure 3. Web demo of our work.

Experimental

Experimental data can be downloaded from here