如何在OpenCV中使用python接口将列表对象分配给cvmat

Question

我想使用cv.SetData()在OpenCV 2.1中使用python接口填充cvMat对象。在C语言中，可以使用下面的代码来完成:

double a[] = {1,2,3,4,5,6,7,8,9,10,11,12};
cvMat Ma;
cvInitMatHeader(&Ma,3,4,CV_64FC1,a,CV_AUTOSTEP);

但是在python接口中，没有InitMatHeader函数，所以我尝试这样编码:

a = range(1,13)
Ma = cv.CreateMatHeader(3,4,cv.CV_64FC1)
cv.SetData(Ma,a,cv.CV_AUTOSTEP)
PrintMat(Ma) # my own function to display all elements in a mat

然后我得到了一个TypeError:CvMat参数'src'没有数据。似乎变量Ma还没有初始化，那么我如何使用numpy来完成这个操作呢?

Answer 1

尝试传递一个array.array('d', vals)或numpy.array(vals, numpy.float64):

import array
import cv
import numpy
ar = array.array('d', range(1, 13))
na = numpy.array(range(1, 13), numpy.float64)
for vals in (ar, na):
    m = cv.CreateMatHeader(3, 4, cv.CV_64FC1)
    cv.SetData(m, vals, cv.CV_AUTOSTEP)
    print cv.mGet(m, 0, 2)

输出:

3.0
3.0