我正在与Keras合作开发CNN网络
我有一个定义了调用Keras模型的preprocess_input
函数(VGG16,VGG19等)的函数here
所以我正在做类似
preprocess_input (inputs)
但是,我想测试几个模型,导致函数中的preprocess_input
不同。我该如何在我的自制函数中说我想使用与正在测试的模型相对应的preprocess_input
?
我曾考虑过将模型的名称作为函数的参数,但是后来,我不知道如何将该字符串转换为允许调用的方式
modelThatIwant.preprocess_input(...)
感谢帮助
答案 0 :(得分:1)
实际上,答案很简单:您只需要在创建的自制函数中添加参数fct
,然后传递与要训练的模型相对应的足够的preprocessing_inputs
。 。感谢@Matias Valdenegro