首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >更改VGG16应用程序的输入张量形状

更改VGG16应用程序的输入张量形状
EN

Stack Overflow用户
提问于 2017-01-28 05:10:48
回答 3查看 13.6K关注 0票数 3

我想将形状(160,320,3)的图像馈送到

代码语言:javascript
运行
复制
 VGG16(input_tensor=input_tensor, include_top=False)   

如何包含将图像重塑为VGG16模型所期望的形状的层,即(224,224,3)?

EN

Stack Overflow用户

发布于 2018-02-12 21:23:10

您需要做两件事:

  1. 通过为图像宽度和高度定义None来显式声明输入形状具有可变大小的输入。
  2. 不使用flatten(),因为它依赖于固定的输入形状。取而代之的是使用GlobalMaxPooling,它不仅可以进行自适应池化,还可以使FC的输入张量扁平化。

我希望这能帮助你实现你想要的东西。

票数 0
EN
查看全部 3 条回答
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/41903051

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档