opencv实现最大池化

对一些稀疏矩阵进行预处理。
利用最大池化来提取重要特征。

import cv2
import numpy as np

def max_pooling2d(image, kernel_size, stride):
    height, width = image.shape[:2]
    kh, kw = kernel_size
    sh, sw = stride
    h_out = (height - kh) // sh + 1
    w_out = (width - kw) // sw + 1
    pooled_image = np.zeros((h_out, w_out), dtype=np.uint8)
    for i in range(0, h_out):
        for j in range(0, w_out):
            start_h = i * sh
            start_w = j * sw
            patch = image[start_h:start_h+kh, start_w:start_w+kw]
            pooled_image[i, j] = np.max(patch)
    return pooled_image

# 读取图像
image = cv2.imread('input_image.jpg', cv2.IMREAD_GRAYSCALE)

# 定义池化核的大小和步幅
kernel_size = (2, 2)
stride = (2, 2)

# 进行最大池化
max_pooled_image = max_pooling2d(image, kernel_size, stride)

# 显示原始图像和最大池化后的图像
cv2.imshow('Original Image', image)
cv2.imshow('Max Pooled Image', max_pooled_image)
cv2.waitKey(0)
cv2.destroyAllWindows()