Question

与this question here相同，但Python除外，不是C ++。

我在OpenCV中有一条错误信息正在打印，即使我做了一个，除了捕捉它。

solution to the linked question建议使用重定向功能，但this comment表明Python for OpenCV中不存在该功能。

如何在打印时停止打印OpenCV错误消息，同时仍允许我打印出我想要的内容？

Answer 1

在编写这个答案时（OpenCV 3.4.1是最后发布的版本），没有办法只过滤默认错误处理程序的输出（我能想到），也没有办法改变错误处理程序。

然而，你的问题让我思考 - 在 highgui 模块中，我们已经有几个函数可以让我们为鼠标，轨迹栏和按钮事件设置Python回调，所以我们可以从中获取灵感代码并修补此新功能。

让我们使用3.4.1版。感兴趣的文件是modules/python/src2/cv2.cpp。我们将从功能OnMouse和pycvSetMouseCallback中获取灵感。

让Python错误处理程序具有与C++ error handler匹配的签名：

error_handler([int]status, [str]func_name, [str]err_msg, [str]file_name, [int]line, [obj]userdata) -> None

我们还添加了对重置为默认错误处理程序的支持 - highgui函数尚未执行此操作。

首先，我们需要一个静态错误处理函数，它将从C ++到Python的参数编组，并调用相应的Python函数来处理错误。就像我们从中获取灵感的函数一样，我们将使用用户数据参数来保存由函数对象和可选的Python用户数据组成的元组。

static int OnError(int status, const char *func_name, const char *err_msg, const char *file_name, int line, void *userdata)
{
    PyGILState_STATE gstate;
    gstate = PyGILState_Ensure();

    PyObject *o = (PyObject*)userdata;
    PyObject *args = Py_BuildValue("isssiO", status, func_name, err_msg, file_name, line, PyTuple_GetItem(o, 1));

    PyObject *r = PyObject_Call(PyTuple_GetItem(o, 0), args, NULL);
    if (r == NULL) {
        PyErr_Print();
    } else {
        Py_DECREF(r);
    }

    Py_DECREF(args);
    PyGILState_Release(gstate);

    return 0; // The return value isn't used
}

接下来，我们需要编写函数来实现Python和C ++之间的绑定。但是，由于我的suspicions of potential memory leaks功能我们会从中获取灵感，我们会做一些补充来解决这个问题 - 我们会跟踪最新的用户数据对象和必要时取消引用它。

// Keep track of the previous handler parameter, so we can decref it when no longer used
static PyObject* last_on_error_param = NULL;

static PyObject *pycvRedirectError(PyObject*, PyObject *args, PyObject *kw)
{
    const char *keywords[] = { "on_error", "param", NULL };
    PyObject *on_error;
    PyObject *param = NULL;

    if (!PyArg_ParseTupleAndKeywords(args, kw, "O|O", (char**)keywords, &on_error, &param))
        return NULL;

    if ((on_error != Py_None) && !PyCallable_Check(on_error))  {
        PyErr_SetString(PyExc_TypeError, "on_error must be callable");
        return NULL;
    }
    if (param == NULL) {
        param = Py_None;
    }

    if (last_on_error_param) {
        Py_DECREF(last_on_error_param);
        last_on_error_param = NULL;
    }

    if (on_error == Py_None) {
        ERRWRAP2(redirectError(NULL));        
    } else {
        last_on_error_param = Py_BuildValue("OO", on_error, param);
        ERRWRAP2(redirectError(OnError, last_on_error_param));
    }
    Py_RETURN_NONE;
}

最后，我们必须注册我们的"special method"。与我们从灵感中获取的功能不同，我们不依赖于某些GUI，我们希望它始终可用，因此我们将代码修改为如下所示：

static PyMethodDef special_methods[] = {
  {"redirectError", (PyCFunction)pycvRedirectError, METH_VARARGS | METH_KEYWORDS, "redirectError(onError [, param]) -> None"},
#ifdef HAVE_OPENCV_HIGHGUI
  {"createTrackbar", pycvCreateTrackbar, METH_VARARGS, "createTrackbar(trackbarName, windowName, value, count, onChange) -> None"},
  {"createButton", (PyCFunction)pycvCreateButton, METH_VARARGS | METH_KEYWORDS, "createButton(buttonName, onChange [, userData, buttonType, initialButtonState]) -> None"},
  {"setMouseCallback", (PyCFunction)pycvSetMouseCallback, METH_VARARGS | METH_KEYWORDS, "setMouseCallback(windowName, onMouse [, param]) -> None"},
#endif
  {NULL, NULL},
};

现在我们可以重建OpenCV，安装它并测试它。我编写了以下脚本来演示功能：

import cv2

def verbose_error_handler(status, func_name, err_msg, file_name, line, userdata):
    print "Status = %d" % status
    print "Function = %s" % func_name
    print "Message = %s" % err_msg
    print "Location = %s(%d)" % (file_name, line)
    print "User data = %r" % userdata

def silent_error_handler(status, func_name, err_msg, file_name, line, userdata):
    pass


print "** Default handler"
try:
    cv2.imshow("", None) # This causes an assert
except cv2.error as e:
    pass

print "** Using verbose handler"
cv2.redirectError(verbose_error_handler, 42)
try:
    cv2.imshow("", None) # This causes an assert
except cv2.error as e:
    pass

print "** Using silent handler"
cv2.redirectError(silent_error_handler)
try:
    cv2.imshow("", None) # This causes an assert
except cv2.error as e:
    pass

print "** Back to default handler"
cv2.redirectError(None)
try:
    cv2.imshow("", None) # This causes an assert
except cv2.error as e:
    pass

使用我的修补版OpenCV（基于上述说明），我在控制台上获得以下输出：

** Default handler
OpenCV(3.4.1) Error: Assertion failed (size.width>0 && size.height>0) in cv::imshow, file F:\GitHub\opencv\modules\highgui\src\window.cpp, line 364
** Using verbose handler
Status = -215
Function = cv::imshow
Message = size.width>0 && size.height>0
Location = F:\GitHub\opencv\modules\highgui\src\window.cpp(364)
User data = 42
** Using silent handler
** Back to default handler
OpenCV(3.4.1) Error: Assertion failed (size.width>0 && size.height>0) in cv::imshow, file F:\GitHub\opencv\modules\highgui\src\window.cpp, line 364

我开始撰写此答案时的一个想法是，default error handler使用以下格式字符串将所有这些邮件输出到stderr

"OpenCV(%s) Error: %s (%s) in %s, file %s, line %d"

我们可以挂钩stderr流，并过滤掉以常量前缀开头的所有行。唉，我无法做到这一点。也许其他人可以吗？

编辑：Patched合并到opencv:master并进行了一些小的修改（主要是我们删除了用户数据参数，因为它在Python中是不必要的）。

如何在Python

1 个答案: