zi2zi: Master Chinese Calligraphy with Conditional Adversarial Networks
Learning eastern asian language typefaces with GAN. zi2zi(字到字, meaning from character to character) is an application and extension of the recent popular pix2pix model to Chinese characters.
Details could be found in this blog post.
Compare with Ground Truth
Brush Writing Fonts
Cursive Script (Requested by SNS audience)
Random Gaussian Style
How to Use
Download tons of fonts as you please
- Python 2.7
- Tensorflow >= 1.0.1
- numpy >= 1.12.1
- scipy >= 0.18.1
To avoid IO bottleneck, preprocessing is necessary to pickle your data into binary and persist in memory during training.
First run the below command to get the font images:
python font2img.py --src_font=src.ttf --dst_font=tgt.otf --charset=CN --sample_count=1000 --sample_dir=dir --label=0 --filter=1 --shuffle=1
Four default charsets are offered: CN, CN_T(traditional), JP, KR. You can also point it to a one line file, it will generate the images of the characters in it. Note, filter option is highly recommended, it will pre sample some characters and filter all the images that have the same hash, usually indicating that character is missing. label indicating index in the category embeddings that this font associated with, default to 0.
After obtaining all images, run package.py to pickle the images and their corresponding labels into binary format:
python package.py --dir=image_directories --save_dir=binary_save_directory --split_ratio=[0,1]
After running this, you will find two objects train.obj and val.obj under the save_dir for training and validation, respectively.
experiment/ └── data ├── train.obj └── val.obj
Create a experiment directory under the root of the project, and a data directory within it to place the two binaries. Assuming a directory layout enforce bettet data isolation, especially if you have multiple experiments running.
To start training run the following command
python train.py --experiment_dir=experiment --experiment_id=0 --batch_size=16 --lr=0.001 --epoch=40 --sample_steps=50 --schedule=20 --L1_penalty=100 --Lconst_penalty=15
schedule here means in between how many epochs, the learning rate will decay by half. The train command will create sample,logs,checkpoint directory under experiment_dir if non-existed, where you can check and manage the progress of your training.
Infer and Interpolate
After training is done, run the below command to infer test data:
python infer.py --model_dir=checkpoint_dir/ --batch_size=16 --source_obj=binary_obj_path --embedding_ids=label[s] of the font, separate by comma --save_dir=save_dir/
Also you can do interpolation with this command:
python infer.py --model_dir= checkpoint_dir/ --batch_size=10 --source_obj=obj_path --embedding_ids=label[s] of the font, separate by comma --save_dir=frames/ --output_gif=gif_path --interpolate=1 --steps=10 --uroboros=1
It will run through all the pairs of fonts specified in embedding_ids and interpolate the number of steps as specified.
Pretained model can be downloaded here which is trained with 27 fonts, only generator is saved to reduce the model size. You can use encoder in the this pretrained model to accelerate the training process.
Code derived and rehashed from:
- pix2pix-tensorflow by yenchenlin
- Domain Transfer Network by yunjey
- ac-gan by buriburisuri
- dc-gan by carpedm20
- origianl pix2pix torch code by phillipi