inference/core/utils/validate.py

def get_num_classes_from_model_prediction_shape(len_prediction, masks=0):
    num_classes = len_prediction - 5 - masks
    return num_classes