它在C ++中的工作方式

Question

在OpenCV 3.4.2中，添加了用于为HoughLines（）返回的每一行返回投票数（累加器值）的选项。在python中，这似乎也受支持，也可以从我的OpenCV安装的python文档字符串中读取：

“每行用2或3元素矢量（ρ，θ）或（ρ，θ< / em>，投票）。”

它也包含在docs中（格式有误）。 但是我找不到在Python中返回3元素选项（ρ，θ，投票）的方法。 这是演示问题的代码：

import numpy as np import cv2 print('OpenCV should be at least 3.4.2 to test: ', cv2.__version__) image = np.eye(10, dtype='uint8') lines = cv2.HoughLines(image, 1, np.pi/180, 5) print('(number of lines, 1, output vector dimension): ', lines.shape) print(lines)

输出

OpenCV should be at least 3.4.2 to test: 3.4.2 (number of lines, 1, output vector dimension): (3, 1, 2) [[[ 0. 2.3212879]] [[ 1. 2.2340214]] [[-1. 2.4609141]]]

所需的行为是增加一列，每行获得的投票数。使用表决值，可以应用比标准阈值更高级的选项，因此经常在SE（here，here，here和{{3} }，有时与HoughCircles（）等效。但是，问题和答案（例如修改源代码和重新编译）均来自正式添加之前的内容，因此不适用于当前情况。

Answer 1

从香草OpenCV 3.4.3开始，您不能在Python中使用此功能。

它在C ++中的工作方式

首先在implementation of HoughLines中，我们可以看到选择输出数组lines的类型的代码：

int type = CV_32FC2;
if (lines.fixedType())
{
    type = lines.type();
    CV_CheckType(type, type == CV_32FC2 || type == CV_32FC3, "Wrong type of output lines");
}

然后我们可以在填充lines时在implementation of HoughLinesStandard中使用此参数：

if (type == CV_32FC2)
{
    _lines.at<Vec2f>(i) = Vec2f(line.rho, line.angle);
}
else
{
    CV_DbgAssert(type == CV_32FC3);
    _lines.at<Vec3f>(i) = Vec3f(line.rho, line.angle, (float)accum[idx]);
}

可以看到类似的代码in HoughLinesSDiv。

基于此，我们需要传入固定类型的_OutputArray，并在3个通道中存储32位浮点数。如何制作固定类型（但不固定大小，因为算法需要调整大小）_OutputArray？让我们再次看看implementation：

通用cv::Mat不是固定类型，cv::UMat也不是
一个选项是std::vector<cv::Vec3f>
另一个选择是cv::Mat3f（即cv::Matx<_Tp, m, n>）

示例代码：

#include <opencv2/opencv.hpp>

int main()
{
    cv::Mat image(cv::Mat::eye(10, 10, CV_8UC1) * 255);

    cv::Mat2f lines2;
    cv::HoughLines(image, lines2, 1, CV_PI / 180, 4); // runs the actual detection
    std::cout << lines2 << "\n";

    cv::Mat3f lines3;;
    cv::HoughLines(image, lines3, 1, CV_PI / 180, 4); // runs the actual detection
    std::cout << lines3 << "\n";

    return 0;
}

控制台输出：

[0, 2.3212879;
 1, 2.2340214;
 -1, 2.4609141]
[0, 2.3212879, 10;
 1, 2.2340214, 6;
 -1, 2.4609141, 6]

Python包装器的工作方式

让我们看看包装HoughLines函数的自动生成的代码：

static PyObject* pyopencv_cv_HoughLines(PyObject* , PyObject* args, PyObject* kw)
{
    using namespace cv;

    {
    PyObject* pyobj_image = NULL;
    Mat image;
    PyObject* pyobj_lines = NULL;
    Mat lines;
    double rho=0;
    double theta=0;
    int threshold=0;
    double srn=0;
    double stn=0;
    double min_theta=0;
    double max_theta=CV_PI;

    const char* keywords[] = { "image", "rho", "theta", "threshold", "lines", "srn", "stn", "min_theta", "max_theta", NULL };
    if( PyArg_ParseTupleAndKeywords(args, kw, "Oddi|Odddd:HoughLines", (char**)keywords, &pyobj_image, &rho, &theta, &threshold, &pyobj_lines, &srn, &stn, &min_theta, &max_theta) &&
        pyopencv_to(pyobj_image, image, ArgInfo("image", 0)) &&
        pyopencv_to(pyobj_lines, lines, ArgInfo("lines", 1)) )
    {
        ERRWRAP2(cv::HoughLines(image, lines, rho, theta, threshold, srn, stn, min_theta, max_theta));
        return pyopencv_from(lines);
    }
    }
    PyErr_Clear();

    // Similar snippet handling UMat...

    return NULL;
}

总结起来，它尝试将在lines参数中传递的对象转换为cv::Mat，然后以cv::HoughLines作为输出参数调用cv::Mat 。（如果失败，那么它将与cv::UMat尝试相同的操作。）不幸的是，这意味着无法为cv::HoughLines赋予固定类型lines，从3.4.3版本开始Python无法访问该功能。

解决方案

据我所知，唯一的解决方案包括修改OpenCV源代码并重建。

Quick Hack

这很简单，编辑implementation of cv::HoughLines并将默认类型更改为CV_32FC3：

int type = CV_32FC3;

但是，这意味着您将始终获得选票（这也意味着OpenCL优化（如果存在）将不会被使用）。

更好的补丁

添加具有默认值return_votes的可选布尔参数false。修改代码，使得当return_votes为true时，type被强制为CV_32FC3。

Header：

CV_EXPORTS_W void HoughLines( InputArray image, OutputArray lines,
                              double rho, double theta, int threshold,
                              double srn = 0, double stn = 0,
                              double min_theta = 0, double max_theta = CV_PI,
                              bool return_votes = false );

Implementation：

void HoughLines( InputArray _image, OutputArray lines,
                 double rho, double theta, int threshold,
                 double srn, double stn, double min_theta, double max_theta,
                 bool return_votes )
{
    CV_INSTRUMENT_REGION()

    int type = CV_32FC2;
    if (return_votes)
    {
         type = CV_32FC3;
    }
    else if (lines.fixedType())
    {
        type = lines.type();
        CV_CheckType(type, type == CV_32FC2 || type == CV_32FC3, "Wrong type of output lines");
    }
    // the rest...

Answer 2

有一个新的python绑定（opencv 4.5.1）

文档：cv.HoughLinesWithAccumulator

OpenCV + python：从3.4.2开始的HoughLines累加器访问

2 个答案:

它在C ++中的工作方式

Python包装器的工作方式

解决方案

Quick Hack

更好的补丁