编写一段图像卷积的代码

这是一段 Python 代码，用于实现图像卷积：

import numpy as np

def convolve(image, kernel):
    # 获取图像和卷积核的尺寸
    image_rows, image_cols, _ = image.shape
    kernel_rows, kernel_cols, _ = kernel.shape
    
    # 计算卷积后的图像大小
    output_rows = image_rows - kernel_rows + 1
    output_cols = image_cols - kernel_cols + 1
    
    # 初始化卷积后的图像
    output = np.zeros((output_rows, output_cols), dtype=np.uint8)
    
    # 对每个像素进行卷积操作
    for i in range(output_rows):
        for j in range(output_cols):
            output[i, j] = np.sum(image[i:i+kernel_rows, j:j+kernel_cols] * kernel)
    
    return output

其中，`image` 是输入的图像，`kernel` 是卷积核，`output` 是卷积后的图像。这个函数首先计算了卷积后的图像大小，然后对每个像素进行卷积操作，最后返回卷积后的图像。