python - 使用 2D 滤波器卷积 3D 图像
问题描述
我有一个形状的图像img.shape = (500, 439, 3)
卷积函数是
def convolution(image, kernel, stride=1, pad=0):
n_h, n_w, _ = image.shape
f = kernel.shape[0]
kernel = np.repeat(kernel[None,:], 3, axis=0)
n_H = int(((n_h + (2*pad) - f) / stride) + 1)
n_W = int(((n_w + (2*pad) - f) / stride) + 1)
n_C = 1
out = np.zeros((n_H, n_W, n_C))
for h in range(n_H):
vert_start = h*stride
vert_end = h*stride + f
for w in range(n_W):
horiz_start = w*stride
horiz_end = w*stride + f
for c in range(n_C):
a_slice_prev = image[vert_start:vert_end,
horiz_start:horiz_end, :]
s = np.multiply(a_slice_prev, kernel)
out[h, w, c] = np.sum(s, dtype=float)
return out
我想在对图像应用任何内核/过滤器后查看图像,所以我得到了以下内容
img = plt.imread('cat.png')
kernel = np.arange(25).reshape((5, 5))
out2 = convolution(img, kernel)
plt.imshow(out2)
plt.show()
我明白了
s = np.multiply(a_slice_prev,内核)
ValueError:操作数无法与形状一起广播 (5,5,3) (3,5,5)
解决方案
np.multiply 正在进行元素乘法。但是,您的论点没有匹配的尺寸。您可以使用它转置您的内核或映像以确保它可以工作:
kernel = kernel.transpose()
您可以在np.multiply
致电之前执行此操作。
推荐阅读
- google-analytics-api - 选择正确的 Google Analytics API 指标
- swift - 如何删除 Firestore 字典值?
- google-sheets - 如何在 Google 表格中使用变量范围?
- python - Opencv houghCircles 未检测到图像中的所有圆圈
- blueprism - Blueprism 6.5 版队列过滤器
- sql - 在 Oracle SQL Developer 中比较不同长度的 VARCHAR2
- excel - 我需要帮助用这个旧宏的新函数替换 Application.Find
- javascript - 应用程序中谷歌表格的错误日期格式
- mysql - 寻找 Woocommerce 批量定价数量范围数据库表
- css - Why LESS css width results in MORE visual width?