当前位置：首页 > 文章列表 > Golang > Go问答 > 使用 GoCV 探测黄色圆形对象

使用 GoCV 探测黄色圆形对象

来源：stackoverflow 2024-03-12 13:00:27 0浏览收藏

哈喽！今天心血来潮给大家带来了《使用 GoCV 探测黄色圆形对象》，想必大家应该对Golang都不陌生吧，那么阅读本文就都不会很困难，以下内容主要涉及到，若是你正在学习Golang，千万别错过这篇文章~希望能帮助到你！

问题内容

我正在尝试检测带有各种颜色的复选标记的圆圈。我想要检测的是带有白色复选标记的黄色，以及未来带有 x 标记的红色圆圈。

到目前为止我一直在尝试的是下面的这些代码。这个使用霍夫圆方法。

package main

import (
    "fmt"
    "image"
    "image/color"
    "os"

    "gocv.io/x/gocv"
)

func detectcircles() {
    if len(os.args) < 2 {
        fmt.println("how to run:\n\tfind-circles [imgfile]")
        return
    }

    filename := os.args[1]

    grey_window := gocv.newwindow("grey window")
    grey_window.movewindow(600, 0)
    window := gocv.newwindow("detected circles")
    defer grey_window.close()
    defer window.close()

    original_img := gocv.imread(filename, gocv.imreadcolor)
    defer original_img.close()

    img := gocv.newmat()
    defer img.close()

    gocv.cvtcolor(original_img, &img, gocv.colorbgrtogray)

    gocv.medianblur(img, &img, 5)

    circles := gocv.newmat()
    defer circles.close()

    gocv.houghcircleswithparams(
        img,
        &circles,
        gocv.houghgradient,
        1, // dp
        // float64(img.rows()/8), // mindist
        26,
        200, // param1
        48,  // param2
        0,   // minradius
        0,   // maxradius
    )

    green := color.rgba{0, 255, 0, 0}
    red := color.rgba{255, 0, 0, 0}

    for i := 0; i < circles.cols(); i++ {
        v := circles.getvecfat(0, i)
        // if circles are found
        if len(v) > 2 {
            x := int(v[0])
            y := int(v[1])
            r := int(v[2])

            gocv.circle(&original_img, image.pt(x, y), r, green, 2)
            gocv.circle(&original_img, image.pt(x, y), 2, red, 3)
        }
    }

    for {
        window.imshow(original_img)

        if window.waitkey(10) >= 0 {
            break
        }
    }
}

通过一些调整，我设法检测到一些像这样的更大的圆圈。

但是当我尝试使用较小的圆圈时仍然没有运气。

这些是我编写的用于隔离颜色的代码。我所做的是在某些 hsv 范围内添加一些颜色遮罩。

package main

import (
    "fmt"
    "image"
    "image/color"
    "os"

    "gocv.io/x/gocv"
)

func DetectCircles2() {
    if len(os.Args) < 2 {
        fmt.Println("How to run:\n\tgo run . [imgfile]")
        return
    }

    filename := os.Args[1]

    window := gocv.NewWindow("detected colors")
    window2 := gocv.NewWindow("end result")
    window3 := gocv.NewWindow("lower bound")
    window4 := gocv.NewWindow("upper bound")
    window2.MoveWindow(600, 0)
    window3.MoveWindow(0, 600)
    window4.MoveWindow(600, 600)
    defer window.Close()
    defer window2.Close()
    defer window3.Close()
    defer window4.Close()

    img := gocv.IMRead(filename, gocv.IMReadColor)
    defer img.Close()

    original_image := img.Clone()
    defer original_image.Close()

    gocv.MedianBlur(img, &img, 3)

    hsv_img := gocv.NewMat()
    defer hsv_img.Close()

    // yellow := gocv.NewScalar(0, 255, 255, 0)
    // yellow_mat := gocv.NewMatFromScalar(yellow, gocv.MatTypeCV8UC3)
    // gocv.CvtColor(yellow_mat, &yellow_mat, gocv.ColorBGRToHSV)
    // hsv := gocv.Split(yellow_mat)
    // fmt.Printf("H: %d S: %d V: %d\n", hsv[0].GetUCharAt(0, 0), hsv[1].GetUCharAt(0, 0), hsv[2].GetUCharAt(0, 0))

    gocv.CvtColor(img, &hsv_img, gocv.ColorBGRToHSV)
    img_rows, img_cols := hsv_img.Rows(), hsv_img.Cols()

    // lb1 := gocv.NewMatWithSizeFromScalar(gocv.NewScalar(0.0, 208.0, 94.0, 0.0), img_rows, img_cols, gocv.MatTypeCV8UC3)
    // ub1 := gocv.NewMatWithSizeFromScalar(gocv.NewScalar(179.0, 255.0, 255.0, 0.0), img_rows, img_cols, gocv.MatTypeCV8UC3)
    lb1 := gocv.NewMatWithSizeFromScalar(gocv.NewScalar(20.0, 50.0, 50.0, 0.0), img_rows, img_cols, gocv.MatTypeCV8UC3)
    ub1 := gocv.NewMatWithSizeFromScalar(gocv.NewScalar(40.0, 255.0, 255.0, 0.0), img_rows, img_cols, gocv.MatTypeCV8UC3)

    lb2 := gocv.NewMatWithSizeFromScalar(gocv.NewScalar(155.0, 100.0, 100.0, 0.0), img_rows, img_cols, gocv.MatTypeCV8UC3)
    ub2 := gocv.NewMatWithSizeFromScalar(gocv.NewScalar(180.0, 255.0, 255.0, 0.0), img_rows, img_cols, gocv.MatTypeCV8UC3)

    lower_bound := gocv.NewMat()
    upper_bound := gocv.NewMat()
    color_isolated_img := gocv.NewMat()
    circles := gocv.NewMat()
    defer lower_bound.Close()
    defer upper_bound.Close()
    defer color_isolated_img.Close()
    defer circles.Close()

    gocv.InRange(hsv_img, lb1, ub1, &lower_bound)
    gocv.InRange(hsv_img, lb2, ub2, &upper_bound)

    gocv.AddWeighted(lower_bound, 1.0, upper_bound, 1.0, 0.0, &color_isolated_img)
    gocv.GaussianBlur(color_isolated_img, &color_isolated_img, image.Pt(9, 9), 2, 2, gocv.BorderDefault)

    gocv.HoughCirclesWithParams(
        color_isolated_img,
        &circles,
        gocv.HoughGradient,
        1,
        float64(color_isolated_img.Rows()/8),
        100,
        20,
        0,
        0,
    )

    green := color.RGBA{0, 255, 0, 0}

    for i := 0; i < circles.Cols(); i++ {
        v := circles.GetVecfAt(0, i)
        fmt.Println(v)
        if len(v) > 2 {
            x := int(v[0])
            y := int(v[1])
            r := int(v[2])

            gocv.Circle(&original_image, image.Pt(x, y), r, green, 2)
        }
    }

    for {
        // window.IMShow(color_isolated_img)
        window2.IMShow(original_image)
        window3.IMShow(lower_bound)
        window4.IMShow(upper_bound)

        if window.WaitKey(10) >= 0 {
            break
        }
    }
}

这是结果。上面的代码只能检测到一个红色圆圈（应该是两个）和一个黄色圆圈。

我一直在尝试转换用 python 编写的轮廓检测，结果却陷入了令人困惑的 gocv api。我标记 gocv 和 opencv 希望这两个社区能够帮助我正确检测彩色小圆圈。任何建议将被认真考虑。非常感谢。