Art-free SAM
收藏Art-Free Generative Models: Art Creation Without Graphic Art Knowledge
数据集概述
该数据集包含多个子数据集,用于支持艺术生成模型的训练和推理。主要数据集包括:
Art-free SAM
-
来源: SA-1B 数据集和 SAM-LLaVA-Captions10M 数据集。
-
结构:
sam_dataset ├── captions │ ├── 0.txt │ ├── 1.txt │ └── ... ├── images │ ├── sa_000000 │ ├── 0.jpg │ ├── 1.jpg │ └── ... │ ├── sa_000001 │ ├── 0.jpg │ ├── 1.jpg │ └── ... │ ├── ... │ └── sa_000999 └──
-
下载: shell cd data python download.py filtered_sam cd ..
Artistic Style Dataset
- 来源: 从 WikiArt 获取的艺术风格数据集。
- 下载: shell cd data python download.py -d art_styles cd ..
Laion-pop500
- 来源: 包含 500 张带注释的图像。
- 下载: shell cd data python download.py -d laion_pop500 cd ..
模型训练与推理
训练艺术适配器
-
命令: shell python train_artistic.py --style_folder <style_folder> --save_path <save_path>
-
示例: shell python train_artistic.py --style_folder data/Art_styles/andre-derain/fauvism/subset1 --save_path <save_path>
推理与生成
-
命令: shell python inference.py --lora_weights <lora_location> --from_scratch --start_noise -1 --infer_prompts <prompts-or-file> --save_dir <save_location>
-
示例: shell python inference.py --lora_weights <lora_location> --from_scratch --start_noise -1 --infer_prompts "Sunset over the ocean with waves and rocks" --save_dir <save_location>
引用
bibtex @misc{ren2024art-free, title={Art-Free Generative Models: Art Creation Without Graphic Art Knowledge}, author={Hui Ren and Joanna Materzynska and Rohit Gandikota and David Bau and Antonio Torralba}, year={2024}, eprint={2412.00176}, archivePrefix={arXiv}, primaryClass={cs.CV}, url={https://arxiv.org/abs/2412.00176}, }




