SWIG Python C ++输出数组给出“未知类型”错误

Question

我正在尝试使用Swig包装一些C ++代码以将numpy数组传递回python。 我正在跟踪一些我在网上看到的使用numpy.i的示例。 这是我的代码的样子。

我将其用作类头文件中的函数定义：

bool grabFrame(int buf_size, unsigned char *buf);

在我的界面文件中，我有：

/* File : OV4682Interface.i */
%module OV4682Interface
%include "std_string.i"

%{
    #define SWIG_FILE_WITH_INIT
    #include "OV4682FrameGrabber.h"
%}

%include "numpy.i"
%init %{
    import_array();
%}

%apply (int DIM1, unsigned char* ARGOUT_ARRAY1) {(int buf_size, unsigned char *buf)};

%include "../inc/OV4682FrameGrabber.h"

我的Python代码如下所示：

import numpy as np
import OV4682Interface as ov

width = 672
height = 380
buf_size = width*height*2

buf = np.zeros(buf_size, dtype=np.uint8)

grab = ov.OV4682FrameGrabber()
grab.grabFrame(buf)

运行此命令时，出现以下错误：

追溯（最近一次通话最后一次）：在grab.grabFrame（buf）中的文件“ OV4682FrameGrabberTest.py”，第44行，在grabFrame def handleFrame中的文件“ /home/ubuntu/rgb_ir_frame_grabber/build/lib/OV4682Interface.py”，第117行（self，* args）：返回_OV4682Interface.OV4682FrameGrabber_grabFrame（self，* args）TypeError：预期为Int尺寸。 给出“未知类型”。

由于某种原因，我收到一个错误消息，说传入的数组的类型未知，但是我已将dtype显式设置为np.uint8。 我想知道是否有人可以指出我在这里做错的事情，因为我有些困惑。

Answer 1

看来我对这篇文章的答案不够仔细（ SWIG + c + Python：传递和接收c数组 ）。 我没有注意到swig将输出数组添加到返回列表中，只是想要传入的数组大小。

我将方法定义更改为：

void grabFrame(int buf_size, unsigned char *buf);

我将python调用更改为：

buf = grab.grabFrame(np.shape(buf)[0])

现在可以正常工作并返回我想要的数据数组。