第2关：实现池化层的反向传播

最新推荐文章于 2024-06-15 21:36:42 发布

一辈子的拥抱

最新推荐文章于 2024-06-15 21:36:42 发布

阅读量340

点赞数

分类专栏： python 文章标签：深度学习 python cnn

本文链接：https://blog.csdn.net/weixin_50719427/article/details/124746847

版权

第1关：实现卷积层的反向传播

import numpy as np
from utils import im2col, col2im
class Convolution:
    def __init__(self, W, b, stride=1, pad=0):
        r'''
        卷积层的初始化

        Parameter:
        - W: numpy.array, (C_out, C_in, K_h, K_w)
        - b: numpy.array, (C_out)
        - stride: int
        - pad: int
        '''
        self.W = W
        self.b = b
        self.stride = stride
        self.pad = pad

        self.x = None
        self.col = None
        self.col_W = None

        self.dW = None
        self.db = None

    def forward(self, x):
        r'''
        卷积层的前向传播

        Parameter:
        - x: numpy.array, (B, C, H, W)

        Return:
        - y: numpy.array, (B, C', H', W')
             H' = (H - Kh + 2P) / S + 1
             W' = (W - Kw + 2P) / S + 1
        '''
        FN, C, FH, FW = self.W.shape