Golang图像处理：如何实现图片的边缘检测

package main

import (
    "fmt"
    "image"
    "image/color"
    "image/jpeg"
    "log"
    "os"
)

func main() {
    // 读取图片文件
    file, err := os.Open("input.jpg")
    if err != nil {
        log.Fatal(err)
    }
    defer file.Close()

    // 解码图片
    img, err := jpeg.Decode(file)
    if err != nil {
        log.Fatal(err)
    }

    // 创建输出图片
    bounds := img.Bounds()
    grayImg := image.NewGray(bounds)

    // 遍历图片每个像素点进行边缘检测
    for x := 1; x < bounds.Max.X-1; x++ {
        for y := 1; y < bounds.Max.Y-1; y++ {
            // 获取3x3邻域内的像素值
            px00 := color.GrayModel.Convert(img.At(x-1, y-1)).(color.Gray).Y
            px01 := color.GrayModel.Convert(img.At(x-1, y)).(color.Gray).Y
            px02 := color.GrayModel.Convert(img.At(x-1, y+1)).(color.Gray).Y
            px10 := color.GrayModel.Convert(img.At(x, y-1)).(color.Gray).Y
            px11 := color.GrayModel.Convert(img.At(x, y)).(color.Gray).Y
            px12 := color.GrayModel.Convert(img.At(x, y+1)).(color.Gray).Y
            px20 := color.GrayModel.Convert(img.At(x+1, y-1)).(color.Gray).Y
            px21 := color.GrayModel.Convert(img.At(x+1, y)).(color.Gray).Y
            px22 := color.GrayModel.Convert(img.At(x+1, y+1)).(color.Gray).Y

            // 计算Sobel算子
            gx := px00 + 2*px01 + px02 - px20 - 2*px21 - px22
            gy := px00 + 2*px10 + px20 - px02 - 2*px12 - px22
            g := gx*gx + gy*gy
            grayImg.SetGray(x, y, color.Gray{255 - uint8(g/64)})
        }
    }

    // 创建输出文件
    outFile, err := os.Create("output.jpg")
    if err != nil {
        log.Fatal(err)
    }
    defer outFile.Close()

    // 编码输出图片
    err = jpeg.Encode(outFile, grayImg, nil)
    if err != nil {
        log.Fatal(err)
    }

    fmt.Println("边缘检测完成！")
}