Keras VGGnet预训练模型可变尺寸Inpu

2024-04-20 08:46:18 发布

您现在位置:Python中文网/ 问答频道 /正文

我想用VGG预训练模型提取368x368大小的图像的特征。根据文档,VGGnet接受224x224大小的图像。有没有办法给Keras VGG提供可变大小的输入?在

这是我的代码:

^{1}$

编辑的代码(有效!)在

^{pr2}$

Tags: 代码文档模型图像编辑特征kerasvgg
1条回答
网友
1楼 · 发布于 2024-04-20 08:46:18

输入大小影响完全连接(Dense)层中神经元的数量。因此,您需要创建自己的完全连接层。在

使用include_top=False调用VGG19删除完全连接的层,然后自己添加它们。检查this code以供参考。在

相关问题 更多 >