如何制作适应模型的通用函数？

Posted 2021-05-04

我正在与Keras开发CNN网络

我有一个函数，我定义了我称之为Kez模型的preprocess_input函数（VGG16，VGG19，...）here

所以我正在做类似的事情

preprocess_input (inputs)

但是，我想测试几个模型，在我的函数中产生不同的preprocess_input。我怎么能在我的自制函数中说我想使用与我正在测试的模型相对应的preprocess_input？

我想把我的模型的名称作为我的函数的参数但是，我不知道如何将这个字符串转换为允许调用的方式

modelThatIwant.preprocess_input(...)

感谢帮助

答案

实际上，答案很简单：你只需要在你创建的自制函数中添加一个参数fct然后，传递对应于你想要训练的模型的足够的preprocessing_inputs ...感谢@Matias Valdenegro