Skip to content

Commit

Permalink
Update documentation
Browse files Browse the repository at this point in the history
  • Loading branch information
tobecwb committed Nov 18, 2023
1 parent 752afa7 commit 3be6d69
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,17 +1,19 @@
## YOLOx8 model segmentation to face features
## YOLOx8 model segmentation for face features

Segment the following features (class):
Segment the following class:

- eye
- eyebrown
- nose
- mouth

![woman_sample](https://raw.githubusercontent.com/tobecwb/facial-features-yolo8x-seg/main/woman_sample.jpg)
---

Segmentation model specially made to generate image masks for manual editing of images from https://github.com/tobecwb/stable-diffusion-regularization-images

The training was done only with female faces, but initial tests showed that it works reasonably well with male faces too.
![man_sample](https://raw.githubusercontent.com/tobecwb/facial-features-yolo8x-seg/main/man_sample.jpg)

For better results, it is interesting to train with images of a male face too (I will do this in the future).

Expand Down

0 comments on commit 3be6d69

Please sign in to comment.