This is the official implementation of the AAAI 2021: AAAI Conference on Artificial Intelligence accepted paper "Sketch Generation with Drawing Process Guided by Vector Flow and Grayscale".
Our paper's official AAAI version is available at: https://ojs.aaai.org/index.php/AAAI/article/view/16140
Our Supplementary Material (PDF) is available at Baidu Netdisk (百度网盘) https://pan.baidu.com/s/1ZZEmpuPXRNBPvG0WHbbPKA. The extraction code (提取码) is 1234
.
An interesting H5 demo using our codes by Xinhua News Agency (新华社) can be found at https://mp.weixin.qq.com/s/v9WooHpZiqo2uVgl2htxeA
Here we give some instructions for running our code.
Zhengyan Tong (童峥岩), 此论文idea提供者、代码原作者、论文主笔者,主要从事计算机视觉方面的研究。发表此论文时为上海交通大学电子信息与电气工程学院信息工程专业大四在读本科生。联系方式: 418004@sjtu.edu.cn
We give three examples that can be run directly (the hyperparameters of these three examples have been fixed).
python cat.py
python dog.py
python girl.py
To draw arbitrary input: python process_order.py
. Of course you need to adjust the following parameters.
input_path = './input/your file'
Input image pathoutput_path = './output'
Do not change thisn = 10
Gray-scale quantization orderperiod = 5
Line(stroke) widthdirection = 10
Direction quantization orderFreq = 100
Save the drawing process every Freq
strokes are drawndeepen = 1
Edge map's intensity. The bigger, the darker.transTone = False
Do not change thiskernel_radius = 3
Edge tangent flow kernel size, do not change thisiter_time = 15
Edge tangent flow kernel iterations times, do not change thisbackground_dir = None
Whether fix the drawing direction in the background, this value could be None
or an integer between (0~180)
CLAHE = True
Whether input uses CLAHE (Do not change this)edge_CLAHE = True
Whether edge map uses CLAHE (Do not change this)draw_new = True
Do not change thisrandom_order = False
Use random drawing order if True
ETF_order = True
Use the drawing order described in our paper if True
process_visible = True
Whether show the drawing process In our supplementary material (PDF), we explain these hyperparameters in more detail and we show more comparisons with existing pencil drawing algorithms. We also offer more
results of our method. Our Supplementary Material is available at Baidu Netdisk (百度网盘) https://pan.baidu.com/s/1ZZEmpuPXRNBPvG0WHbbPKA. The extraction code (提取码) is 1234
.
@inproceedings{tong2021sketch,
title={Sketch generation with drawing process guided by vector flow and grayscale},
author={Tong, Zhengyan and Chen, Xuanhong and Ni, Bingbing and Wang, Xiaohang},
booktitle={Proceedings of the AAAI Conference on Artificial Intelligence},
volume={35},
number={1},
pages={609--616},
year={2021}
}