Official repo for Artist: Aesthetically Controllable Text-Driven Stylization without Training
Artist is a training-free text-driven image stylization method. You give an image and input a prompt describing the desired style, Artist give you the stylized image in that style. The detail of the original image and the style you provide is harmonically integrated.
No need to train, no need to download any ControNets or LoRAs. Just use a pretrained Stable Diffusion.
🔥Sep 21. Add config file for playground-v2 (experimental).
🔥Jul 22. The paper and inference code is released.
🔥Jul 30. Updated huggingface demo, thanks for fffiloni!
Clone the repository:
git clone https://github.com/songrise/Artist
Create a virtual environment and install dependencies:
conda create -n artist python=3.8
conda activate artist
pip install -r requirements.txt
For the first time you execute the code, you need to wait for the download of the Stable Diffusion model from the Hugging Face repository.
Run the following command to start the gradio interface:
python injection_main.py --mode app
Visit http://localhost:7860 in your browser to access the interface.
Notice that for some input image you may need to adjust the parameters to have the best result.
You can also run the following command to stylize an image in the command line:
python injection_main.py --mode cli --image_dir data/example/1.png --prompt "A B&W pencil sketch, detailed cross-hatching" --config example_config.yaml
Aside from the Stable Diffusino model 2.1, we now provide a config file for the playground-v2 model, located in ./example_config_playground.yaml. Note that this feature is still experimental. Compared with SD 2.1, it can have better performance on some image/prompt pairs, but it may also have worse performance on some other pairs. Some good examples are shown below:
@misc{jiang2024artist,
title={Artist: Aesthetically Controllable Text-Driven Stylization without Training},
author={Ruixiang Jiang and Changwen Chen},
year={2024},
eprint={2407.15842},
archivePrefix={arXiv},
primaryClass={cs.CV}
}