Change number of classes

DrSleep / tensorflow-deeplab-resnet

DeepLab-ResNet rebuilt in TensorFlow

MIT License

1.25k stars 431 forks source link

Change number of classes #217

Open vicgabusa opened 2 years ago

vicgabusa commented 2 years ago

Hello! Does anyone know what parts of the code must be modified to change the number of classes? Currently is set to 21...what do I have to do to set it to another number? For example 3 classes? Thanks in advance