How to define number of classes ?

[TrinhThiBaoAnh]

i saw in the train.py : num_classes = 5
Does this consider background as a class?
I have a binary segmentation task, should i put num_classes = 2?

[amlarraz]

yes! you should consider the background as an additional class