OpenCV三角形检测

Question

下面是在图像中查找可能的正方形的工作方法。有什么建议吗，我可以改变这个来代替三角形?

static void getSquares(const Mat& image, vector<vector<Point>>& triangles)
{
squares.clear();
Mat pyr, timg, gray0(image.size(), CV_8U), gray;
pyrDown(image, pyr, Size(image.cols/2, image.rows/2));
pyrUp(pyr, timg, image.size());
vector<vector<Point> > contours;
int planes = 1;
int canny = 0;
if (accuracy) {
    planes = 4;
    canny = 1;
}
for (int c = 0; c < planes; c++)
{
    int ch[] = {c, 0};
    mixChannels(&timg, 1, &gray0, 1, ch, 1);
    for (int l = 0; l < N; l++) {
        if (l == 0 && canny == 1) {
            Canny(gray0, gray, 0, thresh, 5);
            dilate(gray, gray, Mat(), Point(-1,-1));
        } else gray = gray0 >= (l+1)*255/N;
        findContours(gray, contours, CV_RETR_LIST, CV_CHAIN_APPROX_SIMPLE);
        vector<Point> approx;
        for (size_t i = 0; i < contours.size(); i++) {
            approxPolyDP(Mat(contours[i]), approx, arcLength(Mat(contours[i]), true)*0.02, true);
            if(approx.size() == 4 && fabs(contourArea(Mat(approx))) > 1000 && isContourConvex(Mat(approx))) {
                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 < tolerance) squares.push_back(approx);
            }
        }
    }
}

}

任何帮助将不胜感激，否则，如果你能建议另一种方法。

问候,c .

Answer 1

至少你需要改变

approx.size() == 4

approx.size() == 3