首页 > 解决方案 > 预训练的 CNN (MobileNet V2) 输出形状

问题描述

我是 CNN 的新手。我正在为我的项目使用预训练模型。我使用链接作为参考:https ://www.tensorflow.org/tutorials/images/transfer_learning

当我输入:“base_model 摘要”时,我将最后一层的输出形状视为(None,5,5,1280),所以当我的输入尺寸为(160,160,3)时,我期望(1,5,5,1280)输出形状. 但是当我检查时,当输入为 (160,160,3) 时,我看到输出尺寸类似于 (32,5,5,1280),我无法理解 32 的来源以及为什么它不包含在最后一层输出形状中。谁能帮我解决这个问题?

标签: pythonconv-neural-network

解决方案


推荐阅读