Golang は画像のハフ変換と画像セグメンテーションを実装します

import (
    "image"
    "image/color"
    "image/png"
    "math"
    "os"
)

func houghTransform(img image.Image) [][]int {
    bounds := img.Bounds()
    width, height := bounds.Max.X, bounds.Max.Y

    // 初始化霍夫空间
    maxRho := int(math.Sqrt(float64(width*width + height*height)))
    houghSpace := make([][]int, 180)
    for i := range houghSpace {
        houghSpace[i] = make([]int, maxRho*2)
    }

    // 遍历图像的每一个像素点
    for x := 0; x < width; x++ {
        for y := 0; y < height; y++ {
            c := color.GrayModel.Convert(img.At(x, y)).(color.Gray)
            if c.Y > 128 {
                // 如果像素点的灰度值大于阈值，进行霍夫变换
                for theta := 0; theta < 180; theta++ {
                    rho := int(float64(x)*math.Cos(float64(theta)*math.Pi/180) + float64(y)*math.Sin(float64(theta)*math.Pi/180))
                    houghSpace[theta][rho+maxRho]++
                }
            }
        }
    }

    return houghSpace
}

func segmentImage(img image.Image, houghSpace [][]int, threshold int) image.Image {
    bounds := img.Bounds()
    width, height := bounds.Max.X, bounds.Max.Y

    out := image.NewRGBA(bounds)

    // 遍历图像的每一个像素点
    for x := 0; x < width; x++ {
        for y := 0; y < height; y++ {
            c := color.GrayModel.Convert(img.At(x, y)).(color.Gray)
            if c.Y > 128 {
                // 如果像素点的灰度值大于阈值，根据所属的曲线进行分割
                for theta := 0; theta < 180; theta++ {
                    rho := int(float64(x)*math.Cos(float64(theta)*math.Pi/180) + float64(y)*math.Sin(float64(theta)*math.Pi/180))
                    if houghSpace[theta][rho+len(houghSpace[theta])/2] > threshold {
                        out.Set(x, y, color.RGBA{255, 255, 255, 255})
                        break
                    }
                }
            }
        }
    }

    return out
}

func main() {
    // 读入原始图像
    file, _ := os.Open("input.png")
    defer file.Close()
    img, _ := png.Decode(file)

    // 进行霍夫变换
    houghSpace := houghTransform(img)

    // 进行图像分割
    out := segmentImage(img, houghSpace, 100)

    // 保存结果图像
    outFile, _ := os.Create("output.png")
    defer outFile.Close()
    png.Encode(outFile, out)
}