WuuG / MultiStyleTransfer

Implement[Multi-style Generative Network for Real-time Transfer](https://arxiv.org/pdf/1703.06953.pdf) by pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Multi Style Transfer

使用pytorch对论文《Multi-style Generative Network for Real-time Transfer》的复现

数据集分别使用COCO和WikiArt

Below picture shows the training-process on tensorboard:

Loss

Destination Style Image:

Style Image

Original Content Images:

Content Image

Result:

Transfer Image

后面试了下pytorch mobile,模型没有经过任何其他优化,直接使用torch.jit.trace生成.pt,丢进android里,速度比较慢。

Result:

Demo Demo Demo

Acknowledgement

The code benefits from outstanding prior work and their implementations including:

About

Implement[Multi-style Generative Network for Real-time Transfer](https://arxiv.org/pdf/1703.06953.pdf) by pytorch


Languages

Language:Python 100.0%