MobileNet depthmultiplier参数

时间:2018-12-19 16:34:45

标签: tensorflow keras conv-neural-network

我对depth-multiplier (alpha)

中的tf.keras.layers.SeparableConv2D参数感到困惑

tf.keras.layers.DepthwiseConv2D

根据原始论文,M个输入通道将是alpha*M,其中alpha在[0,1]中。我的问题是如何删除频道?我猜它像是辍学吗?感谢澄清

1 个答案:

答案 0 :(得分:0)

这是关于输出形状的。 depth_multiplier是应用于每个输入通道的许多过滤器。这是一个整数正数,因此,如果您有3个通道和depth_multiplier == 4,则在深度卷积之后,您将获得12个通道(3个通道中的每一个都有4个过滤器)

来自docs

  

深度卷积输出通道总数等于filters_in * depth_multiplier

如果要对所有通道应用过滤器,而不是对每个通道单独应用,则需要a regular convolution而不是深度方法。