ATTENTION: IF YOU HAVE DOWNLOADED MODELS BEFORE ALREADY - RE-DOWNLOAD THEM PLEASE. I SHOULD HAVE UPDATED THEM WITH NO DILL VERSION(EXCEPT MODEL THAT WAS CHOSEN AS NEW VERSION UPDATE, SO YOU CAN RECEIVE NOTIFICATIONS.

If you are having issues:

1. Activate your venv

2. pip install dill

3. pip install --upgrade ultralytics

This should fix generic problems.

Small segmentation model aiming to help with head inpainting/enhance.
Note: It is not very accurate, and will struggle with long hair.

This version is smallest available, compromise in accuracy, but it's just 6mb. Medium(50mb) brings in up a notch, but don't think i will be releasing that one in particular.Oke, medium is up.

Here are validation results on ~470 images:

Can also be downloaded at HF(whenever i can actually log in): https://huggingface.co/Anzhc/Anzhcs_YOLOs

Script im using: https://github.com/Anzhc/Training-script-for-Ultralytics-YOLO

Usage

Put into `automatic1111/models/adetailer`, or any other place that you know will work with YOLOv8 models.

Use.

What it does and what is it for?

Head segmentation, duh.

It tries to select head, including hair, while avoiding rest. It is hard to avoid rest...
Model is very inconsistent with long hair.

Overall it should be able to work with wide variety of styles and mediums, as i tries to create a robust enough dataset with quite a bit of abstract, traditional, real and digital content. It is not a big one though, so there is a large room for improvement.

But what you might need it for? To have some fun, while reducing tedium a bit.

P.S. Those inpaints were done on older version, which had a bit worse masking.
All inpaints were done directly with adetailer in single pass with inpaint model.

And it took a few tries, hard to get 7 inpaints right in a row xd