https://github.com/cyberagentailab/multipalette
Implementation of Color Recommendation for Vector Graphic Documents based on Multi-Palette Representation, WACV 2023
https://github.com/cyberagentailab/multipalette
Last synced: 9 months ago
JSON representation
Implementation of Color Recommendation for Vector Graphic Documents based on Multi-Palette Representation, WACV 2023
- Host: GitHub
- URL: https://github.com/cyberagentailab/multipalette
- Owner: CyberAgentAILab
- Created: 2022-09-12T13:02:38.000Z (over 3 years ago)
- Default Branch: main
- Last Pushed: 2023-08-31T06:39:42.000Z (almost 3 years ago)
- Last Synced: 2025-09-10T07:42:51.902Z (9 months ago)
- Language: Jupyter Notebook
- Homepage:
- Size: 8.76 MB
- Stars: 9
- Watchers: 0
- Forks: 1
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# Color recommendation for vector graphic documents
Official implementation of [Color Recommendation for Vector Graphic Documents based on Multi-Palette Representation, WACV 2023](https://openaccess.thecvf.com/content/WACV2023/papers/Qiu_Color_Recommendation_for_Vector_Graphic_Documents_Based_on_Multi-Palette_Representation_WACV_2023_paper.pdf)
## Prerequisites
- Python:3.8
- Poetry: 1.2.*
## Setup
Install requirements and run jupyter.
```
poetry install
poetry run jupyter lab
```
## Quick demo
`notebooks/recomm_colors.ipynb`: recommend colors for multiple palettes in a design
- Trained model of color prediction are in trained_models/.
- Json files for test are pre-created in data/model_test_input/crello_samples/.
You can train a color model on a notebook `notebooks/train_model.ipynb`. We recommended GPU resources to train this model (e.g. Tesla T4 * 1).
You can also create a json file for test from crello dataset on a notebook `notebooks/create_json_file.ipynb`.
## Data
`data/training_data/metadata_colors`: extracted color palettes for Image-SVG-Text elements from [Crello-dataset-v1](https://storage.googleapis.com/ailab-public/canvas-vae/crello-dataset-v1.zip) ([the lastest Crello-dataset](https://github.com/CyberAgentAILab/canvas-vae/blob/main/docs/crello-dataset.md))
`data/training_data/data_bert/data_color`: color corpus of train, validation, and test dataset, and color vocabulary from train dataset
`data/trained_models`: trained model for color recommendation
`model_test_input`: json sample files for testing the results of color recommendation
## Citation
```bibtex
@misc{Qiu_2022,
author = {Qiu Qianru, Wang Xueting, Otani Mayu, and Iwazaki Yuki},
title = {Color Recommendation for Vector Graphic Documents based on Multi-Palette Representation},
doi = {10.48550/ARXIV.2209.10820},
url = {https://arxiv.org/abs/2209.10820},
year = {2022},
}
```