Model Not Predicting Mask Properly

[vprahalad]

Hi, when I am following this tutorial on my own dataset, the training predictions are not properly predicting a mask. What could be the root of this problem? Thanks!

[ianhi]

Hi @vprahalad from looking at your "True mask" it seems you are using a continuous variable for your mask. For this model we used categorical variables - e.g. 0 for background, 1 for mother, 2 for daughter. If your problem allows for it then I recommend changing the true mask to be values of just 0 or 1. You can achieve this by thresholding the image.

If that doesn't work for your problem you may need to adjust the neural net architecture, however the augmentation pipeline and other utilities may still be of help to you.