Force the number of predicted classes

[pkhateri]

Is it possible to force the mask R-CNN model to predict a fixed number of classes and only one instance for each class?
The problem is that only a few of the classes are detected in my test data, while all my images have 5 classes with one instance for each class.