open-mmlab / StyleShot

StyleShot: A SnapShot on Any Style. 一款可以迁移任意风格到任意内容的模型,无需针对图片微调,即能生成高质量的个性风格化图片!
https://styleshot.github.io/
MIT License
229 stars 14 forks source link

Question: Using item[‘prompt’] Instead of item[‘content_prompt’] in Training #3

Open SCUTlihaoyu opened 3 months ago

SCUTlihaoyu commented 3 months ago

I read the training-related code and found that the prompt used in the given training deme is the ‘prompt’ field instead of ‘content_prompt’. Is there an error here? According to the paper, it should be the ‘content_prompt’ without the style, right?

Jeoyal commented 3 months ago

Hey @SCUTlihaoyu , thank you for your interest in our work. The "prompt" here actually refers to "content_prompt". We update the training code accordingly. Additionally, to facilitate training, we have provided a JSON file named stylegallery.jsonl that contains all the training data in a uniform format: "image_file":"", "content_prompt":"", .... You can download it from here. Note that stylegallery.jsonl does not provide "content_image_file" paths, but you can process contours using our tutorial available here.

SCUTlihaoyu commented 3 months ago

Thank you for clarifying my doubts