Ссылка отсутствует (от Canny) ребра

Мне нужно обнаружить все прямоугольники на изображении.

Вот мой код:

Mat PolygonsDetection(Mat src)
 {
    Mat gray;
    cvtColor(src, gray, CV_BGR2GRAY);

    Mat bw;
    Canny(src, bw, 50, 200, 3, true);
    imshow("canny", bw);

    morphologyEx(bw, bw, MORPH_CLOSE, cv::noArray(),cv::Point(-1,-1),1);

    imshow("morph", bw);

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

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

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

        if (approx.size() >= 4 && (approx.size() <= 6))
        {
            int vtc = approx.size();
            vector<double> cos;
            for(int j = 2; j < vtc + 1; j++)
                cos.push_back(Angle(approx[j%vtc], approx[j-2], approx[j-1]));

            sort(cos.begin(), cos.end());

            double mincos = cos.front();
            double maxcos = cos.back();

            if (vtc == 4)// && mincos >= -0.5 && maxcos <= 0.5)
            {
                Rect r = boundingRect(countours[i]);
                double ratio = abs(1 - (double)r.width / r.height);

                line(dst, approx.at(0), approx.at(1), cvScalar(0,0,255),4);
                line(dst, approx.at(1), approx.at(2), cvScalar(0,0,255),4);
                line(dst, approx.at(2), approx.at(3), cvScalar(0,0,255),4);
                line(dst, approx.at(3), approx.at(0), cvScalar(0,0,255),4);
                SetLabel(dst, "RECT", countours[i]);
            }
        }
    }

    return dst;
 }

Вот мой вывод:

введите здесь описание изображения

Вместо 17 прямоугольников (16 маленьких и 1 большой) у меня получилось всего 12 прямоугольников. Я новичок в opencv, может быть, я передаю неправильные параметры функции Canny и morphologyEx... Итак, мои вопросы: что я делаю не так? Как мне его отремонтировать?


person Rougher    schedule 12.04.2014    source источник
comment
Попробуйте поиграться с параметрами Canny; первый и особенно второй порог. Что-то вроде Canny(src, bw, 50, 100, 3, true);, вероятно, приведет вас туда.   -  person HugoRune    schedule 12.04.2014
comment
Это не дубликат. Я проверил это решение, и оно не работает.   -  person Rougher    schedule 12.04.2014


Ответы (1)


Что вы можете сделать, так это использовать обычный трюк с эрозией и расширением. В Matlab для создания маски я обычно делаю следующее. Это делает линии больше и сливаются.

%Create Structuring element
sel = strel('disk', 1);
% bw is the black and white (binary image) created using e.g. Otsu thresholding
bw2 = imcomplement(bw);
bw3 = imerode(bw2, sel);
bw4 = imdilate(bw3, sel);
tightmask = imcomplement(bw4);

Я очень часто использую это для поиска независимых компонентов в изображении.

person Jens Munk    schedule 12.04.2014