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

Question

有没有可能训练一个图像尺寸大于299x299的Tensorflow Inception V3模型？似乎盗梦空间V3 CNN只为这种图像尺寸而设计。

Answer 1

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

卷积层权重通常应该独立于图像大小，因此您可以使用这些权重。具有n个完全连接的节点的预训练网络的FC层将具有大小为[m X n]的权重矩阵。这一层将期望输入到该层的大小为m。但是，由于图像大小的变化，当您从新数据集馈送图像时，最终将得到不同的m值(卷积滤波器对不同的图像大小进行卷积)。

在添加了新的密集层之后，您可以对网络进行微调，以仅在顶层进行训练(保持其下方的卷积块的权重固定)。