Пример #1
0
def get_model(input_shape, nclasses=3):
    if starting_model_fp is None:
        return ku.model_c3(input_shape, nclasses, compile_model=True)
    else:
        return load_model(starting_model_fp)
Пример #2
0
def get_model(input_shape, nclasses=3):
    if starting_model_fp is None and trn_passes != 0:  # not an eval only script
        return ku.model_c3(input_shape, nclasses, compile_model=True)
    else:
        return load_model(starting_model_fp)