如何把OpenCV Python获取的图像传递到C层处理

本文介绍了如何在C/C++环境中处理通过OpenCV Python获取的图像数据。首先,需要将Dynamsoft Barcode Reader和OpenCV的cv2.pyd库引入Python环境。在C层,通过指针获取OpenCV图像数据,并利用这些数据调用C/C++的图像处理库,如Dynamsoft Barcode Reader SDK进行条形码检测。文章提供了构建Python扩展的步骤,以及使用摄像头实时读取和检测条形码的Python代码示例。
摘要由CSDN通过智能技术生成

原文:https://blog.csdn.net/yushulx/article/details/52788051

用OpenCV Python来开发,如果想要用到一些C/C++的图像处理库,就需要创建Python的C/C++扩展,然后把数据从Python传递到底层处理。这里分享下如何在C/C++层获取数据。

参考原文:How to Convert OpenCV Image Data from Python to C

作者:Xiao Ling

翻译:yushulx

开发环境

Python C/C++扩展

把DynamsoftBarcodeReaderx86.dll和cv2.pyd拷贝到目录Python27\Lib\site-packages。

OpenCV Python获取的图像数据类型是numpy.ndarray:

> rval, frame = vc.read();
> print type(frame)
> <type 'numpy.ndarray'>

在C层我们希望能获取到数据的指针。查看OpenCV源码文件opencv\modules\python\src2\cv2.cv.hpp可以找到方法:

PyObject *o;
if (!PyArg_ParseTuple(args, "O", &o))
    return NULL;
 
PyObject *ao = PyObject_GetAttrString(o, "__array_struct__");
PyObject *retval;
 
if ((ao == NULL) || !PyCObject_Check(ao)) {
    PyErr_SetString(PyExc_TypeError, "object does not have array interface");
    return NULL;
}
 
PyArrayInterface *pai = (PyArrayInterface*)PyCObject_AsVoidPtr(ao);
if (pai->two != 2) {
    PyErr_SetString(PyExc_TypeError, "object does not have array interface");
    Py_DECREF(ao);
    return NULL;
}
 
// Construct data with header info and image data 
char *buffer = (char*)pai->data; // The address of image data
int width = pai->shape[1];       // image width
int height = pai->shape[0];      // image height
int size = pai->strides[0] * pai->shape[0]; // image size = stride * height

这样就可以了。现在可以用这个数据做点事情,比如调用barcode接口来做检测。我依然用Dynamsoft Barcode Reader SDK做示例。首先需要构建一下数据:

char *total = (char *)malloc(size + 40); // buffer size = image size + header size
memset(total, 0, size + 40);
BITMAPINFOHEADER bitmap_info = {40, width, height, 0, 24, 0, size, 0, 0, 0, 0};
memcpy(total, &bitmap_info, 40);
 
// Copy image data to buffer from bottom to top
char *data = total + 40;
int stride = pai->strides[0];
for (int i = 1; i <= height; i++) {
    memcpy(data, buffer + stride * (height - i), stride);
    data += stride;
}

接下来就可以检测barcode了:

// Dynamsoft Barcode Reader initialization
    __int64 llFormat = (OneD | QR_CODE | PDF417 | DATAMATRIX);
    int iMaxCount = 0x7FFFFFFF;
    ReaderOptions ro = {0};
    pBarcodeResultArray pResults = NULL;
    ro.llBarcodeFormat = llFormat;
    ro.iMaxBarcodesNumPerPage = iMaxCount;
    printf("width: %d, height: %d, size:%d\n", width, height, size);
    int iRet = DBR_DecodeBuffer((unsigned char *)total, size + 40, &ro, &pResults);
    printf("DBR_DecodeBuffer ret: %d\n", iRet);
    free(total); // Do not forget to release the constructed buffer 
     
    // Get results
    int count = pResults->iBarcodeCount;
    pBarcodeResult* ppBarcodes = pResults->ppBarcodes;
    pBarcodeResult tmp = NULL;
    retval = PyList_New(count); // The returned Python object
    PyObject* result = NULL;
    for (int i = 0; i < count; i++)
    {
        tmp = ppBarcodes[i];
        result = PyString_FromString(tmp->pBarcodeData);
        printf("result: %s\n", tmp->pBarcodeData);
        PyList_SetItem(retval, i, Py_BuildValue("iN", (int)tmp->llFormat, result)); // Add results to list
    }
    // release memory
    DBR_FreeBarcodeResults(&pResults);

在Windows上构建Python扩展需要先设置一下,不然会出错。我使用Visual Studio 2015。命令行如下:

SET VS90COMNTOOLS=%VS140COMNTOOLS%
python setup.py build install

好了。现在可以用Python脚本来调用了。首先打开摄像头:

import cv2
from dbr import *
import time
 
vc = cv2.VideoCapture(0)

接下来读取一帧的数据:

cv2.imshow(windowName, frame)
rval, frame = vc.read();

现在可以实时检测barcode了:

initLicense("<license>") # Invalid license is fine.
results = decodeBuffer(frame)
if (len(results) > 0):
    print "Total count: " + str(len(results))
    for result in results:
        print "Type: " + types[result[0]]
        print "Value: " + result[1] + "\n"

 

源码

https://github.com/yushulx/opencv-python-webcam-barcode-reader

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值