caffe SqueezeNet:prototxt中的全连接FC层在哪里

caffe SqueezeNet: where is the fully connected FC layer in prototxt

我正在研究 caffe SqueezeNet prototxt link

我只是想知道FC层在哪里? (我只看到类型:data、conv、relu、pooling、concat、SoftmaxWithLoss和accuracy)

原因是 FC 层有大量参数,在某些架构中占网络参数的大部分。 SqueezeNet 的作者移除了 FC,用一个卷积层和一个全局平均池取而代之。

conv 层的过滤器数量等于 classes 的数量,将前一层的输出处理为(大致)每个 class 的映射。汇集对每个地图的响应进行平均。他们最终得到一个维度等于 classes 数量的扁平向量,然后将其馈送到 SoftMax 层。

通过这些修改(不要忘记他们提出的 Fire 模块),他们能够显着减少内存占用。

我强烈建议您阅读SqueezeNet paper

SqueezeNet 没有全连接层,而是使用全局平均池。