如何在Keras中为Inception V3设置自定义的input_shape?

我知道Inception V3的input_shape(299,299,3)。但是在Keras中,如果include_top设置为False,可以构建具有自定义input_shape的Inception V3版本。

“input_shape: 可选的形状元组,仅当include_topFalse时指定(否则输入形状必须是(299, 299, 3)(使用'channels_last'数据格式)或(3, 299, 299)(使用'channels_first'数据格式)。它应该正好有3个输入通道,宽度和高度不应小于75。例如,(150, 150, 3)就是一个有效的值” – https://keras.io/applications/#inceptionv3

这是如何实现的?为什么只有当include_topfalse时才可以设置自定义的input_shape


回答:

这是可能的,因为该模型完全是卷积的。卷积操作不关心图像的大小,它们是“滑动滤波器”。如果图像很大,输出也会很大;如果图像很小,输出也会很小。(然而,滤波器本身具有由kernel_size定义的固定大小,以及输入和输出滤波器)

当你使用include_top时,你不能这样做,因为这个模型可能在最后使用了Flatten()层 followed by Dense层。Dense层需要固定的输入大小(由Flatten根据图像大小给出),否则无法创建可训练的权重(可变数量的权重没有意义)

Related Posts

在Keras中实现条件批归一化

我正在尝试在Keras中实现条件批归一化。我假设需要创…

Keras: 在LSTM网络中添加MDN层

我的简短问题:给定舞蹈序列的训练数据,以下详细描述的长…

移除最大池化层导致PyTorch中CUDA内存溢出错误

显卡:GTX1070Ti 8GB,批次大小64,输入图…

实现自定义数据集与PyTorch

我正在尝试修改从https://github.com/…

如何使用Google Protobuf解析、编辑和生成object_detection/pipeline.config文件

我在一个常见的集成学习范式中训练多个模型,目前我在处理…

我的GridSearchCV不起作用,我不知道为什么

大家好,我在使用GridSearchCV时遇到了问题,…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注