基于Pytorch的风格迁移的一些结论

2022-09-03  本文已影响0人  赤色要塞满了

代码主要参考Pytorch官方教程,不赘述。
这是内容图:


dancing.jpg

这是风格图:


picasso.jpg

结论

如果弱风格,强内容,那么从内容图开始跑300个epoch,大概200个epoch就有内味了,主要是色调的变化;


weak style 300 epoch.png

如果中弱风格,中强内容,可以从内容图开始跑1000个epoch;


weak style 1000 epoch.png

如果中强风格,中弱内容,就从白噪声跑2000个epoch;


strong style 2000 epoch.png

如果强风格,弱内容,就从白噪声只跑1000个epoch,仔细看的话,还是有一丁点的噪点没被平滑掉。


strong style 1000 epoch.png

逐渐离谱!

问题

  1. 如何复用风格,有说是加一个cnn用于训练,而不是直接训练权重。待测试。
  2. 每个epoch都需要截取clamp权重张量到0~1之间吗?待测试。
上一篇 下一篇

猜你喜欢

热点阅读