How to use the same code for binary segmentation?

[minaahmed]

trafficstars

Hi can I use the same code for binary segmentation

[davidtvs]

You can see here how the number of classes is extracted from a batch returned by the data generator. So I would start by writing a data generator class for your dataset. Each batch returned by the generator should have ground-truth images/labels where the last dimension is the number of classes, in your case, 2.

You would also need to add your new dataset data generator here and here.

Hope that helps.