Image Segmentation, Binary class: Labels?


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 !


Hi Thomas,
Were you able to solve this?

Check out this thread…may help you!