SWIG-python的numpy参数传入C函数

最新推荐文章于 2023-10-16 00:19:54 发布

为了一个愿望

最新推荐文章于 2023-10-16 00:19:54 发布

阅读量1.6k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： python c/c++ swig

本文链接：https://blog.youkuaiyun.com/qq_33766091/article/details/81950871

本文介绍了如何利用SWIG将Python中的numpy矩阵直接传入C语言编写的函数。通过numpy.i文件，SWIG能够将numpy格式转换为C可接受的数组类型。在Ubuntu上，可以安装python-numpy后，使用SWIG和setup.py来编译并运行示例程序。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

SWIG-python的numpy参数传入C函数

简介

由于python需要调用c语言写的函数，且最好直接传入numpy的矩阵。
经查询，numpy直接提供一个numpy.i（swig配置格式）的文件供swig使用，可以将numpy格式转换成C可用的数组格式。

numpy获取地址：https://github.com/numpy/numpy
numpy.i在目录中位置：numpy/tools/swig
numpy在 ubuntu 上安装： apt-get install python-numpy

我的程序如下：
setup.py:

#!/usr/bin/python
from distutils.core import setup, Extension

convolutionquantized_3x3_test_module = 
            Extension('_convolutionquantized_3x3_test',
            library_dirs=['./'],
            libraries=['ncnn'],
            sources=['convolutionquantized_3x3_test_wrap.cxx'] )

setup (name = 'convolutionquantized_3x3_test',
       version = '0.1',
       author      = "lb",
       description = """for test""",
       ext_modules = [convolutionquantized_3x3_test_module],
       py_modules = ["convolutionquantized_3x3_test"],)