使用OpenCV检测红色矩形（或正方形）

Question

我是OpenCv的新手，我需要检测图像中的红色矩形。

我发现了很多示例，例如one，这些示例如何使用OpenCV检测红色圆圈。但是，

如何使用OpenCV和C ++检测红色矩形？

我想做的和上面的教程一样，可以识别这个圆圈：

但是，我要识别矩形和正方形，而不是圆形。我想从最简单的部分开始，接下来我将更改照明条件。

Answer 1

由于您是新手，因此建议您使用最简单的方法来检测形状对象（例如矩形）。首先，您将对片段图像应用阈值处理（可以使用Canny()），然后使用矩形的findContours()提取轮廓，最后使用approxPolyDP()-将轮廓形状近似为另一个形状，但顶点数取决于我们指定的精度。它是Douglas-Peucker算法的实现。

cv::Mat src = ...;
cv::Mat gray = ...;
cv::Mat bw;
cv::Canny(gray, bw, 800, 850, 5, true); // Modify values for your use-case

std::vector<std::vector<cv::Point>> countours;
cv::findContours(bw.clone(), countours, CV_RETR_CCOMP, CV_CHAIN_APPROX_SIMPLE);

std::vector<cv::Point> approx;
cv::Mat dst = src.clone();

for(int i = 0; i < countours.size(); i++)
{
    cv::approxPolyDP(Mat(countours[i]), approx, arcLength(Mat(countours[i]), true) * 0.01, true);


    if (approx.size() == 4)
    {
        double maxCosine = 0;

        for (int j = 2; j < 5; j++)
        {
            double cosine = fabs(angle(approx[j % 4], approx[j - 2], approx[j - 1]));
            maxCosine = MAX(maxCosine, cosine);
        }

        if (maxCosine < 1.2)
        {
            cv::line(dst, approx.at(0), approx.at(1), cvScalar(0,0,255), 4);
            cv::line(dst, approx.at(1), approx.at(2), cvScalar(0,0,255), 4);
            cv::line(dst, approx.at(2), approx.at(3), cvScalar(0,0,255), 4);
            cv::line(dst, approx.at(3), approx.at(0), cvScalar(0,0,255), 4);
        }

    }
}