Modelpublic

a13ph/occlusion-aware-face-segmentation

sourceHugging Faceupdated 7mo agoView on Hugging Face

0likes

Model Card

All thanks and credit to @ototadana.

Mirrored to rid code of antiquated openxlab and openmim dependencies.

license: cc0-1.0 pipeline_tag: image-segmentation tags:

—mmsegmentation
—face
—occlusion ---

Occlusion-aware face segmentation

A model for occlusion-aware face segmentation.

This model was created following the procedures in mmsegmentation's PR [[Feature] Support Delving into High-Quality Synthetic Face Occlusion Segmentation Datasets #2194](https://github.com/open-mmlab/mmsegmentation/pull/2194).

For more information, see:

—https://github.com/open-mmlab/mmsegmentation/pull/2194/files
—https://github.com/kennyvoo/face-occlusion-generation

How to use

Use with mmsegmentation.

Example:

python

from mmseg.apis import inference_model, init_model, show_result_pyplot
import mmcv

config_file = 'deeplabv3plus_r101_512x512_face-occlusion.py'
checkpoint_file = 'deeplabv3plus_r101_512x512_face-occlusion-93ec6695.pth'
model = init_model(config_file, checkpoint_file, device='cuda:0')

img = 'face-image.png'
result = inference_model(model, img)
show_result_pyplot(model, img, result, show=True, out_file='result.jpg', opacity=0.5)