Function convolve2d

cnn_class/custom_blur.py:40–49

(X, W)

# same size as input
def convolve2d(X, W):
    n1, n2 = X.shape
    m1, m2 = W.shape
    Y = np.zeros((n1 + m1 - 1, n2 + m2 - 1))
    for i in range(n1):
        for j in range(n2):
            Y[i:i+m1,j:j+m2] += X[i,j]*W
    ret = Y[m1//2:-m1//2+1,m2//2:-m2//2+1]
    assert(ret.shape == X.shape)
    return ret

# smaller than input
# def convolve2d(X, W):