如何使通用函数适应模型?

时间:2018-10-05 09:11:05

标签: python keras conv-neural-network

我正在与Keras合作开发CNN网络

我有一个定义了调用Keras模型的preprocess_input函数(VGG16,VGG19等)的函数here

所以我正在做类似

preprocess_input (inputs)

但是,我想测试几个模型,导致函数中的preprocess_input不同。我该如何在我的自制函数中说我想使用与正在测试的模型相对应的preprocess_input

我曾考虑过将模型的名称作为函数的参数,但是后来,我不知道如何将该字符串转换为允许调用的方式

modelThatIwant.preprocess_input(...)

感谢帮助

1 个答案:

答案 0 :(得分:1)

实际上,答案很简单:您只需要在创建的自制函数中添加参数fct,然后传递与要训练的模型相对应的足够的preprocessing_inputs。 。感谢@Matias Valdenegro