Как объединить соседние строки, полученные в результате HoughLinesP, в OpenCV Android?

Я делаю Android-приложение для обнаружения головоломки судоку и поиска ответа на нее. В первой части мне нужно извлечь судоку из данного изображения. Я успешно смог использовать HoughLinesP, чтобы найти коробку с содержанием (судоку), но из-за толстых линий, присутствующих на изображении, генерируется очень большое количество линий. Мне нужно объединить houghLines, которые принадлежат одной и той же "толстой линии", в одну строку, но я не могу этого сделать. Мой код до сих пор:

    Mat inputMat = new Mat(inputBitmap.getHeight(), inputBitmap.getWidth(), CvType.CV_8UC1);
    Utils.bitmapToMat(inputBitmap, inputMat, false);

    Mat grayMat = new Mat();
    Imgproc.cvtColor(inputMat, grayMat, Imgproc.COLOR_BGR2GRAY);

    Mat blurMat = new Mat();
    Imgproc.blur(grayMat, blurMat, new Size(1, 1));

    Mat cannyEdges = new Mat();
    Imgproc.Canny(blurMat, cannyEdges, 50, 200);

    Mat lines = new Mat();
    Imgproc.HoughLinesP(cannyEdges, lines, 1, Math.PI / 180, 150);

    List<double[]> horizontalLines = new ArrayList<>();
    List<double[]> verticalLines = new ArrayList<>();


    for(int i = 0 ; i < lines.cols() ; i++) {
        double[] line = lines.get(0, i);
        double x1 = line[0];
        double y1 = line[1];
        double x2 = line[2];
        double y2 = line[3];

        if (Math.abs(y2 - y1) < Math.abs(x2 - x1)) {
            horizontalLines.add(line);
        } else if (Math.abs(x2 - x1) < Math.abs(y2 - y1)) {
            verticalLines.add(line);
        }
    }

Итак, технически мне нужно 10 горизонтальных линий и 10 вертикальных линий. PS: видна только судоку, т.е. никаких других линий на изображении нет, только обычная судоку. Пожалуйста, помогите мне решить эту проблему и большое спасибо за ваше время:)

0 ответов

Другие вопросы по тегам