2024-04-20 08:46:18 发布
网友
我想用VGG预训练模型提取368x368大小的图像的特征。根据文档,VGGnet接受224x224大小的图像。有没有办法给Keras VGG提供可变大小的输入?在
这是我的代码:
编辑的代码(有效!)在
输入大小影响完全连接(Dense)层中神经元的数量。因此,您需要创建自己的完全连接层。在
Dense
使用include_top=False调用VGG19删除完全连接的层,然后自己添加它们。检查this code以供参考。在
include_top=False
输入大小影响完全连接(
Dense
)层中神经元的数量。因此,您需要创建自己的完全连接层。在使用
include_top=False
调用VGG19删除完全连接的层,然后自己添加它们。检查this code以供参考。在相关问题 更多 >
编程相关推荐