Image Segmentation, Binary class: Labels?

Hi,

I want to do Image Segmentation. I have images and their targets, which are white&black masks. In fact, there are two areas to classify (it is also binary classification).

Masks are also arrays of values 0 or 255.

Is there a way to tell to our model that “class A = 0” and “class B = 255” ?

Or should I manually resave my masks after replacing 255 by 1 ?

Thanks in advance !

2 Likes

Hi Thomas,
Were you able to solve this?
Regards,


Check out this thread…may help you!