c446951
1
2
3
4
def get_num_classes_from_model_prediction_shape(len_prediction, masks=0): num_classes = len_prediction - 5 - masks return num_classes
