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

About training new painting styles #53

Open hensiesp32 opened 2 years ago

hensiesp32 commented 2 years ago

Hello, your work is very creative, but I found that you didn't have train dataset, so how did you train the strokes of different painting styles? If I want to train a Chinese style painting stroke, what should I do? Hope your reply. Thanks so much!

jiupinjia commented 2 years ago

Hi @hensiesp32, thanks for your questions and sorry for the late response. When training the neural renderer, the dataset is generated on-the-fly. Please see the dataset class (line 95 in utils.py) for more details. If you want to train your own stroke style, you may customize the dataset class in your own way. Thanks for your interest.