search

Yijunmaverick / UniversalStyleTransfer

The source code of NIPS17 'Universal Style Transfer via Feature Transforms'.
MIT License
598 stars 91 forks source link

Using multiple masks #3

Open okdewit opened 6 years ago

okdewit commented 6 years ago

Am I right in assuming the usage of multiple masks which leave part of the picture untransformed (as demonstrated in the (first picture)[https://github.com/Yijunmaverick/UniversalStyleTransfer/blob/master/figs/p3.jpg]) is not implemented in the LUA script?

The command from the readme uses a binary mask with styles for fg/bg, not multiple alpha masks, right?

th test_wct_mask.lua -content YourConentPath -style YourStylePath1,YourStylePath2 -mask YourBinaryMaskPath

This is not a problem as it's fairly easy to edit afterwards, but I'm curious as to whether I'm missing something.

Yijunmaverick commented 6 years ago

@okdewit You're right.

I do not include this in my code. As long as you have the mask, you can decide to only stylize some parts and keep the rest as the same with the content. The example I showed in the first picture with three styles and three masks is obtained by running my model three times.