索贝尔滤波器算法(C++) (无库)

#include <iostream>
#include <cmath>

// 定义索贝尔滤波器算法函数
void sobelFilter(const int* inputImage, int* outputImage, int width, int height) {
    // 定义索贝尔滤波器模板
    int sobelX[3][3] = {{-1, 0, 1}, {-2, 0, 2}, {-1, 0, 1}};
    int sobelY[3][3] = {{-1, -2, -1}, {0, 0, 0}, {1, 2, 1}};

    // 对每个像素点进行滤波计算
    for (int y = 1; y < height - 1; y++) {
        for (int x = 1; x < width - 1; x++) {
            int gx = 0, gy = 0;

            // 计算梯度值
            for (int j = -1; j <= 1; j++) {
                for (int i = -1; i <= 1; i++) {
                    gx += inputImage[(y + j) * width + (x + i)] * sobelX[j + 1][i + 1];
                    gy += inputImage[(y + j) * width + (x + i)] * sobelY[j + 1][i + 1];
                }
            }

            // 计算梯度幅值
            outputImage[y * width + x] = std::sqrt(gx * gx + gy * gy);
        }
    }
}

int main() {
    // 定义输入图像和输出图像
    int inputImage[9] = {1, 2, 1, 0, 0, 0, -1, -2, -1};
    int outputImage[9] = {0};

    // 调用索贝尔滤波器算法函数
    sobelFilter(inputImage, outputImage, 3, 3);

    // 打印输出图像
    for (int i = 0; i < 9; i++) {
        std::cout << outputImage[i] << " ";
    }
    std::cout << std::endl;

    return 0;
}