| 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](https://mmsegmentation.readthedocs.io/en/latest/)'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](https://mmsegmentation.readthedocs.io/en/latest/get_started.html). |
| |
| 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) |
| ``` |
