输入0与图层global_average_pooling2d_4不兼容：预期ndim = 4，发现ndim = 2错误

Question

我正在尝试微调VGG16模型。 我已经删除了最后5层

(*block5_pool (MaxPooling2D),flatten(Flatten),fc1 (Dense),fc2 (Dense),predictions (Dense)*). 

现在，我想添加一个全局平均池层，但出现此错误

输入0与图层global_average_pooling2d_4不兼容：预期ndim = 4，找到的ndim = 2 **

这里似乎是什么问题？

model = VGG16(weights='imagenet', include_top=True)
model.layers.pop()
model.layers.pop()
model.layers.pop()
model.layers.pop()
model.layers.pop()
x = model.output
x = GlobalAveragePooling2D()(x)

Answer 1

如果要删除最后四层，则只需使用include_top=False 。 此外，使用pooling='avg'将GlobalAveragePooling2D图层添加为最后一层：

model = VGG16(weights='imagenet', include_top=False, pooling='avg')

---

关于为什么原始解决方案不起作用的说明：如该答案中已经建议的那样，您不能在模型的layers属性上使用pop()方法来删除层。 相反，您需要直接引用它们的输出（例如， model.layers[-4].output ），然后如果要添加新的连接，则将它们提供给其他层。