3

我卡在我的代码中的某一点。首先简要说明我在做什么：作为输入，有一张地板的图像。使用 Canny 和 HoughLinesP 算法，我想将整面墙分割成许多“小”部分，正如您在此处看到的那样，这同时是理想的输出（这里没有 canny），我想得到- 一段是两条红线之间。

在此处输入图像描述

好吧，因为我实际上在这里得到了这个

在此处输入图像描述

我想知道如何合并彼此太近的线。例如，第 2、4、3、5、6 行应该是一行，也算作一。比第 7 行到第 15 行也应该是一个，这将是第二行。

当然，我做了一些研究并尝试了这个：

#include <opencv/cv.h>
#include <opencv/highgui.h>
#include <opencv2/opencv.hpp>
#include <opencv2/core/core.hpp>
#include <experimental/filesystem>

using namespace cv;
using namespace std;

Mat srcReal = //here's this image http://imgur.com/a/Kcjp6
Mat src, dst, cdst;
vector<Vec4i> lines;

void wallMapping(Mat src) {

  Scalar mu, sigma;
  meanStdDev(src, mu, sigma);

  Canny(src, dst, mu.val[0] - sigma.val[0], mu.val[0] + sigma.val[0], 3, false);

  cvtColor(dst, cdst, CV_GRAY2BGR);
  HoughLinesP(dst, lines, 1, CV_PI / 2, 50, 50, 200);

  sort(lines.begin(), lines.end(), vec4iSortByX());                                 ///sort all lines by number

  for (size_t i = 1; i < lines.size(); i++) {                                   
        Vec4i current = lines[i];                                                   ///set current lines
        Point pt1 = Point(current[0], current[1]);
        Point pt2 = Point(current[2], current[3]);

        Vec4i previous = lines[i - 1];                                              ///set previous lines
        Point ppt1 = Point(previous[0], previous[1]);
        Point ppt2 = Point(previous[2], previous[3]);

        int gradient1, gradient2;
        if (pt1.x - pt2.x != 0) {
            gradient1 = (pt1.y - pt2.y) / (pt1.x - pt2.x);
            gradient2 = (ppt1.y - ppt2.y) / (ppt1.x - ppt2.x);
        }
        Point avrgpt1, avrgpt2;

        if (gradient1 == gradient2) {

            avrgpt1.x = (pt1.x + ppt1.x) / 2;
            avrgpt2.x = (pt2.x + ppt2.x) / 2;

            avrgpt1.y = (pt1.y + ppt1.y) / 2;
            avrgpt2.y = (pt2.y + ppt2.y) / 2;
        }

        double angle = atan2(ppt2.y - ppt1.y, ppt2.x - ppt1.x) * 180.0 / CV_PI;     ///draw only vertical lines (90 degree)
        if (angle) {
            std::vector<int> lineLabel;
            int numLines = cv::partition(lines, lineLabel, isEqual);
            line(cdst, avrgpt1, avrgpt2, Scalar(0, 0, 255), 2, CV_AA);
        }

        cv::putText(cdst, to_string(i+1), pt2 + Point(0, 10), 4, 1, Scalar(0, 255, 0), 1, 8, false);

        //some other stuff
 }

//main
int main(int argc, char *argv[]) {
       wallMapping(srcReal);

       waitKey(0);
       return 0;
}

但是当您查看上图时，您可以看到我在代码中的合并想法不起作用。所以我会很高兴我的代码的想法、方法或更正！谢谢！

4

1 回答 1

0

在单独的二值图像中绘制线条，对它们执行形态闭合，以便彼此靠近的线条合并，然后用一些小内核（例如 3x3）腐蚀图像以留下细线。

之后，从您刚刚创建的二进制图像中获取非零坐标，并在所需图像中绘制这些坐标。

为了让这些新线回到向量中，您可能会在二进制图像中找到轮廓findContours()，然后只找到每个轮廓的 2 个极值点（x 或 y 值的极值点） - 这些是描述线的点。

于 2017-07-25T20:09:40.910 回答