可以训练图像大于299x299的Tensorflow Inception V3模型吗?

时间:2018-02-10 02:11:11

标签: tensorflow

是否可以训练图像尺寸大于299x299的Tensorflow Inception V3模型?似乎Inception V3 CNN仅针对此图像尺寸而设计。

1 个答案:

答案 0 :(得分:0)

只要您不在顶部包含完全连接(密集)图层,就可以使用不同的图像尺寸。 您可以通过在加载模型时添加此参数来实现  base_model = InceptionV3(weights=weights, include_top=False)

卷积层权重通常应与图像大小无关,因此您可以使用这些权重。具有n个完全连接的节点的预训练网络的FC层将具有大小为[m X n]的权重矩阵。此图层将期望该图层的输入大小为m。但是,由于图像大小的变化,当您从新数据集中提取图像时,最终会得到m的不同值(卷积滤镜在不同的图像大小上进行卷积)。

添加新的密集层后,您可以对网络进行微调以单独训练顶层(保持固定下方的转换块的权重)。