def get_model(model_name, num_classes): if model_name == "create_new_model": return create_new_model(num_classes) elif model_name == "AlexNet": return AlexNet(num_classes) elif model_name == "LeNet5": return LeNet5(num_classes) elif model_name == "VGG16": return VGG16(num_classes) elif model_name == "ResNet50": return ResNet50(num_classes) elif model_name == "InceptionV3": return InceptionV3(num_classes) elif model_name == "DeepFace": return DeepFace(num_classes) elif model_name == "VGGFace": return VGGFace(num_classes) elif model_name == "VGGFaceHalf": return VGGFace(num_classes)
def test_model(): print(VGGFace(2).summary())