【DeepLearning学习】1.Python实现卷积操作

思路:

采用纯for循环加list实现

输入数据[[1,2,3],[1,2,3]]是2维的,相当h=2,w=3。
拿2维矩阵卷积来举例,具体思路就是先遍历h,再遍历w,卷积的方式选择是VALID,就是不足卷积核大小的数据就舍弃。
这里说一下VALID模式下输出矩阵大小的计算公式,【(H-K_h+1) / s】 ,这里【】代表向上取整,H代表输入大小,K_h代表卷积核大小,【9.5】等于10.。。。哈哈打不出向上取整的符号。

# 这些库仅做显示使用
from PIL import Image
import matplotlib.pyplot as plt
import numpy as np

def conv3(data, kernel, s):
    channels = len(data)
    rows = len(data[0])
    cols = len(data[0][0])

    H = len(kernel[0])
    W = len(kernel[0][0])

    res = []
    for cha in range(channels):
        lines = []
        for r in range(0, rows-H+1, s):
            line = []
            for c in range(0, cols-W+1, s):
                sums = 0
                for h in range(H):
                    for w in range(W):
                        sums += data[cha][r+h][c+w] * kernel[cha][h][w]
                line.append(sums)
            lines.append(line)
        res.append(lines)
    return res

def conv2(data, kernel, s):
    rows = len(data)
    cols = len(data[0])

    H = len(kernel)
    W = len(kernel[0])

    lines = []
    for r in range(0, rows-H+1, s):
        line = []
        for c in range(0, cols-W+1, s):
            sums = 0
            for h in range(H):
                for w in range(W):
                    sums += data[r+h][c+w] * kernel[h][w]
            line.append(sums)
        lines.append(line)
    return lines

data = [
    [1,1,1],
    [0,0,0],
    [1,1,1],
]
k_2 = [
    [1,1],
    [1,0]
]
k_3 = [
    [1,1,1],
    [1,0,1],
    [1,1,1]
]

res = conv2(data, k_2, s=1)

# 这是针对2维矩阵的卷积
img = Image.open("/home/lfy/Pictures/eg_tulip.jpg")
gray = np.array(img.convert('L'))
res = conv2(gray, k_3, s=2)
res = np.array(res)
plt.figure()
plt.imshow(res)
plt.show()

# 这是针对3维矩阵的卷积
# img = Image.open("/home/lfy/Pictures/eg_tulip.jpg")
# img = np.array(img)
# img = img.transpose((2,0,1))
# res1 = conv3(img, [k_3, k_3, k_3])
# res1 = np.array(res1 )
# res1 = np.array(res1).transpose(1,2,0)
# plt.figure()
# plt.imshow(res1 )
# plt.show()
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值