utils.py 文件源码

def load_model(img_size = (100,100,3),mode = "VGG"):
    '''softmax
        Helper for loading model. Only VGG available now
    '''

    if mode == "VGG":
        input_template = Input(batch_shape=(None,) + img_size)

        model = VGG16(input_tensor=input_template, 
                      weights='imagenet', 
                      include_top=False)
    else:
        raise NotImplemented

    return model