ty625911724 / jittor-face-parsing

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Jittor version: Face parsing

BiSeNet: Bilateral Segmentation Network for Real-time Semantic Segmentation
Changqian Yu, Jingbo Wang, Chao Peng, Changxin Gao, Gang Yu, Nong Sang. In ECCV, 2018.

Notes

We implement the semantic segmentation and make-up for facial images. The official weights are converted to jittor. Download the weights and put them in ./checkpoints/

The Google Drive link: https://drive.google.com/drive/folders/11Z5hTWku3ARkTJOmtnoWpKMqTwKF7V1i?usp=sharing

Quick start

Generate segmentation results for facial images :

python test.py --input_dir ./img/input/ --output_dir ./img/parse/

Make up for facial images:

python make_up.py --input_image ./img/input/458.jpg --output_image ./img/makeup/458.jpg

This example change the hair color. Other parts, such as lips, can also be changed.

Acknowledgements

This repository borrows partially from the original codes and face parasing torch repository.

About


Languages

Language:Python 100.0%