djbielejeski / a-person-mask-generator

Extension for Automatic1111 and ComfyUI to automatically create masks for Background/Hair/Body/Face/Clothes in Img2Img
MIT License
264 stars 34 forks source link

Looking forward to detect head accessories #13

Closed arafatx closed 9 months ago

arafatx commented 10 months ago

Very nice plugin that can Identify faces and hair. Is it possible in the future it can detect head accessories like scarf or hijab

Thank you.

ArnorWing commented 10 months ago

Great extension. Hand detection would be really good too if possible?

djbielejeski commented 10 months ago

Can you give me an example image to test against?

It looks like the model has support for "accessories" that I did not expose in the UI

image

https://developers.google.com/mediapipe/solutions/vision/image_segmenter#multiclass-model