lengstrom / fast-style-transfer

TensorFlow CNN for fast style transfer ⚡🖥🎨🖼
10.91k stars 2.6k forks source link

Custom Trained Styles Are Generating Outputs With Noise #227

Open rooseveltrp opened 4 years ago

rooseveltrp commented 4 years ago

Hi,

Thanks for starting this awesome project. I took a crack at it and trained a custom style. However, after I evaluate it, the output comes out looking like below:

fast_style_noise

Any idea why it's looking like this? Do I need to add more test images? Iterations? Or something else?

Thanks

beinando commented 4 years ago

Hi, i have exactly the same problem. . i tried some hyperparameters but it appears that there is no change of the generated output images. . does somebody have an idea what could be the reason behind this?

rooseveltrp commented 4 years ago

Hi @beinando, I couldn't get it to work with lengstrom's fast-style-transfer. But I had success with https://github.com/hwalsuklee/tensorflow-fast-style-transfer. Make sure when you train the models, the image is .jpg and a square. I had great success with 300 x 300.

beinando commented 4 years ago

Thanks man, you really saved me a lot of time and effort. I was already starting to work on this repository you proposed, but i was not really hopeful since i firstly got the same weird outputs. Did you already test out some hyperparameters? And more importantly, did you also rescale the images from the COCO - Dataset ? Thank you and have a nice day!

rooseveltrp commented 4 years ago

No problem, @beinando! Glad to help. No, I didn't rescale the images from the COCO Dataset. Just the image with the desired style.