thangtran480/hair-segmentation

hair segmentation in mobile device

hair-segmentation's Issues

Use of multiple color

Hello,

Thanks for sharing this. We found your code very useful. Is it possible to have a different color other than purple for hair? Where can I find the exact piece of code for the same?

Regards,
Vidhi

Deploy model to Android with Tensorflow Lite

ml model is not working in android

hi dev.. how to get mask image from this ml model.pls give some advice

How to apply color on Hair

I downloaded the source code, but when I pick an image from the gallery it is taking too much time to respond, and the other thing is there is no option to add color to Hair,
Please guide ...

Data to train

Hello @thangtran480
Your project seems to be very nice!
I've tried to train my own model there is no problem by using Figaro1K, but how did you use lfw images ?
Thanks :)

Is your pretrained model weights(hairnet_matting_30.hdf5) the best model or we need further training?

I need more quality, if I train further, will I get better model weights?

ValueError: zero-size array to reduction operation maximum which has no identity

Hi @thangtran480
Every time I ran python train.py, I jus get the error "ValueError: zero-size array to reduction operation maximum which has no identity".

Train all CelebAMask-HQ photos takes several hours, so I reduced the number of images to identify something wrong, but nothing.

I made data struture like this:

Are there any more steps I need to take?

iOS example would be great, unable to load tflite model on iOS with Tensorflow-lite

Was trying to load the converted tflite model with the example here:
Most are using pod file with 1.13.1 version of Tensorflow-lite.
Currently it looks like 1.14+ is a requirement for your model.
https://github.com/toniz/deeplab-on-ios

Android model?

Do you happen to know when you will release an Android model and demo? That could do hair segmentation in realtime.

Models are missing

Hi,

Thank you for your hard work, but in the commit of
"add train model with GPU
@thangtran480
thangtran480 committed on 6 Jun"

you deleted the folder called models with its files, can you reupload them in the latest version please ? :)

Bad result

result is so bad

You save the pixel in a text file. I set these pixels in a bitmap, but it shows a black screen. How can I solve this problem?

how to prepare image to predict mask

Hello.

As I see model learned on aligned faces.
All faces in center of image and most of it is normalized (see in one direction).

But what we need to do with any photo with face?
Do we need find, rotate, crop area of face before predict?
Or this library do it by themself?

color of hair

I want to classify hair color.

My first naive solution is cv2.mean(image, hair_mask) and find similarity to set of predefined colors.

What do you think about?

Android sample not work

After select an image app will hang and get an anr warning.

Extract hair region

How to extract the hair region from the original image after the hair is segmented? Once we predict the hair segmentation part, we are getting results like the purple color on hair which shows the detection part. but I want only segmented hair (with the original color of hair) and remove face and background. Can you guide me on how to manipulate pixels with OpenCV? @thangtran480 @gungui98