手动实现卷积操作

最新推荐文章于 2024-08-01 16:55:04 发布

飞鱼大魔王

最新推荐文章于 2024-08-01 16:55:04 发布

阅读量1.6k

点赞数

分类专栏： Tensorflow

本文链接：https://blog.csdn.net/weixin_39142651/article/details/80679930

版权

本文介绍了手动实现图像卷积的操作，包括SAME模式下的补0处理和卷积核的应用，详细解析了程序的关键部分。

摘要由CSDN通过智能技术生成

图像卷积操作

程序

import numpy as np
from matplotlib import pyplot as plt
import tensorflow as tf
from io import BytesIO
from PIL import Image
import base64
import pylab



def conv2d(input, filter, stride, padding):
    # batch x height x width x channels
    in_s = input.shape
    # height x width x in_channels x out_channels
    f_s = filter.shape

    '''assert condition:
    ===>
    if not condition:
        raise AssertionError()
    '''
    assert len(in_s) == 4, 'input size rank 4 required!'
    assert len(f_s) == 4, 'filter size rank 4 required!'
    assert f_s[2] == in_s[3], 'intput channels not match filter channels.'
    assert f_s[0] >= stride and f_s[1] >&