提取霍夫线相交坐标并将数据输入到记事本或Excel中

时间:2014-12-15 06:42:57

标签: opencv coordinates intersection lines

我需要帮助来获取HoughLines生成的线条的坐标并将其提取到输出文件(记事本,Excel或任何其他输出文件)。

我设法获得了这些线条,根据我在这个网站上的研究,我发现了一个帖子,告诉我如何获得坐标,但是由于我的理解有限,我无法让代码运行我的原始Hough代码并得到交点指向输出文件。

这是我最初的Hough代码:

#pragma once
#include <C:\OpenCV2.2\include\opencv\cv.h>
#include <C:\OpenCV2.2\include\opencv\highgui.h>
#include <C:\OpenCV2.2\include\opencv2\core\core.hpp>
#include <C:\OpenCV2.2\include\opencv2\imgproc\imgproc.hpp>
#include <C:\OpenCV2.2\include\opencv2\highgui\highgui.hpp>

#include <stdio.h>
#include <math.h>

#include <opencv2/opencv.hpp>
#include <iostream>

using namespace std;
using namespace cv;

int main(int argc, char* argv[])

{

cv::Mat dst_img, gray_img, contour_img, contrast_img;
cv::Mat src_img = cv::imread("C:\\Frame-1.bmp"); //Source image path
dst_img = src_img.clone();
dst_img.convertTo(contrast_img, -1, 1.5, 0);
cv::cvtColor(contrast_img, gray_img, CV_BGR2GRAY);
cv::Canny(gray_img, contour_img, 75, 225, 3);

vector<Vec2f> lines_;

HoughLines(contour_img, lines_, 1, CV_PI/180, 200);

for( size_t i = 0; i < lines_.size(); i++ )
{
float rho = lines_[i][0];
float theta = lines_[i][1];

double a = cos(theta), b = sin(theta);
double x0 = a*rho, y0 = b*rho;

Point pt1(cvRound(x0 + 1000*(-b)),
    cvRound(y0 + 1000*(a)));
Point pt2(cvRound(x0 - 1000*(-b)),
    cvRound(y0 - 1000*(a)));

cv::clipLine(gray_img.size(), pt1, pt2);

if(!dst_img.empty())
    line( dst_img, pt1, pt2, Scalar(0, 0, 255), 1, CV_AA);

cv::imwrite("result.bmp", dst_img);
}

namedWindow("My Image");
imshow("My Image", dst_img);

waitKey(0);

return 0;

}

这是我想要放入原始代码的代码的链接:

I am struck at finding the point of intersection of most lines in an image

现在我的原始代码绘制Houghlines并导出图像(作为result.bmp),同时在新窗口中显示图像。

我只需要知道如何以及在何处放置新代码以及另外的代码来获取坐标到记录板等输出文件的原始数据,最希望与result.bmp在同一文件夹中(名称为输出文件可以是任何东西,只需要它就可以了。)

很抱歉,如果这个问题听起来像是一个初学者的问题(我真的很喜欢),我们非常感谢任何帮助。非常感谢提前。

其他信息:我正在使用OpenCV 2.2和Microsoft Visual Studio Academic 2010

编辑:这是所有三个代码(Hough,坐标提取和将数据导出到记事本),但作为一个完整的初学者,我不知道如何使它们在单个代码中完成。

#pragma once
#include <C:\OpenCV2.2\include\opencv\cv.h>
#include <C:\OpenCV2.2\include\opencv\highgui.h>
#include <C:\OpenCV2.2\include\opencv2\core\core.hpp>
#include <C:\OpenCV2.2\include\opencv2\imgproc\imgproc.hpp>
#include <C:\OpenCV2.2\include\opencv2\highgui\highgui.hpp>

#include <stdio.h>
#include <math.h>

#include <opencv2/opencv.hpp>
#include <iostream>

#define PointMinusPoint(P,Q,R)      {(P).x = (Q).x - (R).x; (P).y = (Q).y - (R).y;}
#define PointCross(P,Q)             (((P).x*(Q).y)-((P).y*(Q).x))
#define SIGN(X)             (((X)>=0)? 1:-1 )
#define ABS(a)              ((a) >= 0 ? (a) : (-(a)))
#define ROUND(a)            ((SIGN(a)) * ( ( int )( ABS(a) + 0.5 ) ) ) 

typedef struct{
   int x,y;
} MYintPOINT;

typedef struct {
    MYintPOINT  pStart;
    MYintPOINT  pEnd;
} MyLine;

using namespace std;
using namespace cv;

int main(int argc, char* argv[])

{

cv::Mat dst_img, gray_img, contour_img, contrast_img;
cv::Mat src_img = cv::imread("C:\\Frame-1.bmp"); //Source image path
dst_img = src_img.clone();
dst_img.convertTo(contrast_img, -1, 1.5, 0);
cv::cvtColor(contrast_img, gray_img, CV_BGR2GRAY);
cv::Canny(gray_img, contour_img, 75, 225, 3);

vector<Vec2f> lines_;

HoughLines(contour_img, lines_, 1, CV_PI/180, 200);

for( size_t i = 0; i < lines_.size(); i++ )
{
float rho = lines_[i][0];
float theta = lines_[i][1];

double a = cos(theta), b = sin(theta);
double x0 = a*rho, y0 = b*rho;

Point pt1(cvRound(x0 + 1000*(-b)),
    cvRound(y0 + 1000*(a)));
Point pt2(cvRound(x0 - 1000*(-b)),
    cvRound(y0 - 1000*(a)));

cv::clipLine(gray_img.size(), pt1, pt2);

if(!dst_img.empty())
line( dst_img, pt1, pt2, Scalar(0, 0, 255), 1, CV_AA);

cv::imwrite("result.bmp", dst_img);
}

int findLinesIntersectionPoint(const MyLine*l1, const MyLine*l2, MYintPOINT *res){
    MYintPOINT  p  = l1->pStart;
    MYintPOINT  dp;
    MYintPOINT  q  = l2->pStart;
    MYintPOINT  dq;
    MYintPOINT  qmp;            // q-p
    int         dpdq_cross;     // 2 cross products
    int         qpdq_cross;     // dp with dq,  q-p with dq
    float       a;

PointMinusPoint(dp,l1->pEnd,l1->pStart);
PointMinusPoint(dq,l2->pEnd,l2->pStart);
PointMinusPoint(qmp,q,p);

dpdq_cross = PointCross(dp,dq);
if (!dpdq_cross){
    // Perpendicular Lines
    return 0;
}

qpdq_cross = PointCross(qmp,dq);
a = (qpdq_cross*1.0f/dpdq_cross);

res->x = ROUND(p.x+a*dp.x);
res->y = ROUND(p.y+a*dp.y);
return 1;
}

string FileName= FileName_S.c_str();
string::size_type Extension = FileName_S.find_last_of('.');                  // Find extension point

Mat mInputImg;
mInputImg= imread(FileName_S,1);
Size szInput= mInputImg.size();
const string DestinationFileName = FileName_S.substr(0, Extension) + "_ImageData.csv";   // Form the new name with container
ofstream myfile (DestinationFileName.c_str());
if (!myfile.is_open())
{
    MessageBox(L"Unable to Open File");
}
string Text= format("Row, Col , Pixel Data,\n");
myfile << Text;

for (int Row = 0; Row < szInput.height; Row++)
{
    for (int Col = 0; Col < szInput.width; Col++)
    {
        string Text= format("%d , %d , %d",Row,Col,mInputImg.at<uchar>(Row,Col));
        myfile << Text;
        myfile << "\n";
    }
}
myfile.close();

namedWindow("My Image");
imshow("My Image", dst_img);

waitKey(0);

return 0;

}

1 个答案:

答案 0 :(得分:1)

将数据导出到记事本或excel文件非常容易。以下是将mat导出到csv文件的代码。使用所需数据格式化String以导出所需数据。

/*Exporting a Mat to Excel(.csv) file*/
    string FileName= FileName_S.c_str();
    string::size_type Extension = FileName_S.find_last_of('.');                  // Find extension point

    Mat mInputImg;
    mInputImg= imread(FileName_S,1);
    Size szInput= mInputImg.size();
    const string DestinationFileName = FileName_S.substr(0, Extension) + "_ImageData.csv";   // Form the new name with container
    ofstream myfile (DestinationFileName.c_str());
    if (!myfile.is_open())
    {
        MessageBox(L"Unable to Open File");
    }
    string Text= format("Row, Col , Pixel Data,\n");
    myfile << Text;

    for (int Row = 0; Row < szInput.height; Row++)
    {
        for (int Col = 0; Col < szInput.width; Col++)
        {
            string Text= format("%d , %d , %d",Row,Col,mInputImg.at<uchar>(Row,Col));
            myfile << Text;
            myfile << "\n";
        }
    }
    myfile.close();