计算卷积层中的输出大小

Question

如果神经网络卷积层的输入是大小为 128X128X3 的图像并对其应用 40 个大小为 5X5 的滤波器，则输出大小是多少？

Answer 1

你可以使用这个公式[(W?K+2P)/S]+1。

• W 是输入音量 - 在你的情况下是 128
• K 是内核大小 - 在您的情况下为 5
• P 是填充 - 在你的情况下 0 我相信
• S 是步幅 - 您没有提供。

所以，我们输入公式：

Output_Shape = (128-5+0)/1+1

Output_Shape = (124,124,40)

注意：如果未提供，Stride 默认为 1，40in(124, 124, 40)是用户提供的过滤器数量。

Answer 2

你可以通过两种方式找到它：简单的方法：input_size - (filter_size - 1)

W - (K-1)
Here W = Input size
            K = Filter size
            S = Stride
            P = Padding

但是第二种方法是找到输出大小的标准。

Second method: (((W - K + 2P)/S) + 1)
        Here W = Input size
        K = Filter size
        S = Stride
        P = Padding