cv :: gpu :: HoughLines无法正常工作

Question

我试图检测一些使用霍夫变换在cv::gpu::GpuMat结构的行。我曾尝试使用gpu::HoughLines和gpu::HoughLinesP，但即使门槛极低，我也没有得到任何结果。在调试过程中，我发现应该包含结果的容器（houghLines）只有零存储在其中。我已经写了下面给出的代码，

static cv::Mat drawHoughLinesOnMat (cv::gpu::GpuMat hough_Mat, cv::gpu::GpuMat houghLines) 
{ 
    cv::Mat output_Mat; 
    cv::cvtColor(cv::Mat(hough_Mat), output_Mat, CV_GRAY2BGR); 

    std::vector<cv::Vec4i> lines_vector; 
    if (!houghLines.empty()) 
    { 
     lines_vector.resize(houghLines.cols); 
     cv::Mat temp_Mat (1, houghLines.cols, CV_8UC3, &lines_vector[0]); 
     houghLines.download (temp_Mat); 
    } 

    for (size_t i=0; i<lines_vector.size(); ++i) 
    { 
     cv::Vec4i l = lines_vector[i]; 
     cv::line(output_Mat, cv::Point(l[0], l[1]), cv::Point(l[2], l[3]), cv::Scalar(0, 0, 255), 1, 8); 
    } 

    return output_Mat; 
} 


int main() 
{ 
    cv::Mat input = cv::imread(INPUT_DATA_1->c_str(), CV_LOAD_IMAGE_GRAYSCALE); 
    std::string imageType = getImgType(input.type()); 

    cv::gpu::GpuMat mat_input(input), bil_out, mat_thresh, hough_lines; 
    cv::gpu::HoughLinesBuf hough_buffer; 

    int bilateral_thresh = 15; // 5 == 0.085s; 15 == 0.467s at run-time 
    cv::gpu::bilateralFilter(mat_input, bil_out, bilateral_thresh, bilateral_thresh*2, bilateral_thresh/2); 
    //cv::gpu::threshold(bil_out, mat_thresh, 10, 255, CV_THRESH_BINARY); 
    cv::gpu::Canny(bil_out, mat_thresh, 10, 60, 5); 

    cv::gpu::HoughLinesP(mat_thresh, hough_lines, hough_buffer, 1.0f, (float)(CV_PI/180.0f), 5, 1); 
    //cv::Mat test_hough(hough_lines); 
    cv::Mat hough_Mat = drawHoughLinesOnMat(mat_input, hough_lines); 
    cv::gpu::HoughLines(mat_thresh, hough_lines, 1.0f, (float)(CV_PI/180.0f), 1, true); 
    /*cv::Mat */hough_Mat = drawHoughLinesOnMat(mat_input, hough_lines); 

    return EXIT_SUCCESS 
} 

我使用的是图像，

有人能告诉我它是什么，我做错了..？提前致谢。！

坎尼滤波器的输出，

编辑：

我已经在HoughLines的CPU版本测试，它似乎工作得很好。

EDIT_2：

@ jet47发布的解决方案非常完美。

Answer 1

您使用不正确的代码下载从GPU返回结果到CPU：

lines_vector.resize(houghLines.cols); 
cv::Mat temp_Mat (1, houghLines.cols, CV_8UC3, &lines_vector[0]); 
houghLines.download (temp_Mat); 

您使用不正确类型temp_Mat - CV_8UC3，它必须是CV_32SC4。

正确的代码是：

lines_vector.resize(houghLines.cols); 
cv::Mat temp_Mat(1, houghLines.cols, CV_32SC4, &lines_vector[0]); 
houghLines.download(temp_Mat); 

Answer 2

我的猜测是你使用的方法已过时（但我不完全确定）。

这是我应该怎样做（如本Example Code证明）：

//d_src filled with your image somewhere 
GpuMat d_lines; 
{ 
    Ptr<cuda::HoughSegmentDetector> hough = cuda::createHoughSegmentDetector(1.0f, (float) (CV_PI/180.0f), 50, 5); 

    hough->detect(d_src, d_lines); 
} 

vector<Vec4i> lines_gpu; 
if (!d_lines.empty()) 
{ 
    lines_gpu.resize(d_lines.cols); 
    Mat h_lines(1, d_lines.cols, CV_32SC4, &lines_gpu[0]); 
    d_lines.download(h_lines); 
} 

for (size_t i = 0; i < lines_gpu.size(); ++i) 
{ 
    Vec4i l = lines_gpu[i]; 
    line(dst_gpu, Point(l[0], l[1]), Point(l[2], l[3]), Scalar(0, 0, 255), 3, LINE_AA); 
} 

编辑以上使用OpenCV的3.0接口