search

jiupinjia / stylized-neural-painting

Official Pytorch implementation of the preprint paper "Stylized Neural Painting", in CVPR 2021.
https://jiupinjia.github.io/neuralpainter/
Creative Commons Zero v1.0 Universal
1.56k stars 262 forks source link

Style Transfer Colab Notebook error #50

Closed metaphorz closed 2 years ago

metaphorz commented 2 years ago

The first part of the image-to-painting-nst.ipynb works fine since this is creating a painting through brush strokes and then a video.

The second part is the style transfer code, and I am getting an error here:

Screen Shot 2021-12-12 at 4 57 40 PM
jiupinjia commented 2 years ago

Hi @metaphorz, thanks for your feedback. It seems your style image isn't loaded correctly. Did you check your arg.style_img_path before loading it?

metaphorz commented 2 years ago

You are right. I had a mistake in the style file spec.