diff --git a/README.md b/README.md index dd1169e..de153ae 100644 --- a/README.md +++ b/README.md @@ -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).