Python 调用 C++ 并传递numpy 数据

最新推荐文章于 2023-03-24 20:40:19 发布
最新推荐文章于 2023-03-24 20:40:19 发布
阅读量1.7k 收藏 2
点赞数
分类专栏： 工程 文章标签： python c++ numpy
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/jsf921942722/article/details/123799915
版权

文章目录

C++ 代码

Demo.h

#pragma once
void GeneratorGaussKernel(int ksize, float sigma, float* kernel);

void LeftAndRightMirrorImageUInt8(unsigned char* in, unsigned char* out, int width, int height);

void LeftAndRightMirrorImageFloat(float* in, float* out, int width, int height);

void UpAndDownMirrorImageFloat(float* in, float* out, int width, int height);

void UpAndDownMirrorImageUInt8(unsigned char* in, unsigned char* out, int width, int height);
void ImageFilterFloat(float* in, float* out, int width, int height, float* filterKernel, int kw, int kh);

void SaltAndPepperFloat(float* in, float* out, int width, int height, float minV, float maxV, float proportion);

void SaltAndPepperUInt8(unsigned char* in, unsigned char* out, int width, int height, float minV, float maxV, float proportion);
void ImageMinMax(float* in, int width, int height, int channels, float* minV, float* maxV);

void ImageMinMax(unsigned char* in, int width, int height, int channels, unsigned char* minV, unsigned char* maxV);
void ImageMulAAddBFloatFloat(float* in, float* out, int width, int height, int channels, float A, float B);

void ImageMulAAddBUInt8UInt8(unsigned char* in, unsigned char* out, int width, int height, int channels, float A, float B);
void ImageMulAAddBUInt8Float(unsigned char* in, float* out, int width, int height, int channels, float A, float B);

void NormalizeUInt8Float(unsigned char* in, float* out, int width, int height, int channels, int type);
void NormalizeFloatFloat(float* in, float* out, int width, int height, int channels, int type);
void RGBAvgUInt8Float(unsigned char* in, float* out, int width, int height);
void RGBAvgFloatFloat(float* in, float* out, int width, int height);

Demo.cpp

#include <Python.h>
#include <malloc.h>
#include <numpy/arrayobject.h>
#include <iostream>
#include <vector>
#include <xmmintrin.h>
#include <immintrin.h>
#include "omp.h"

class ImageCoord {
public:
    ImageCoord() {
        x = 0; y = 0;
    }

    ImageCoord(const ImageCoord& coord) {
        x = coord.x;
        y = coord.y;
    }
    ImageCoord(int x, int y) {
        this->x = x;
        this->y = y;
    }
    void operator= (ImageCoord& coord) {
        x = coord.x;
        y = coord.y;
    }

    int x, y;
};

class Random {
public:
    Random() {
        srand((unsigned int)time(NULL));
    }

    ImageCoord RandomImageCoord(int width, int height) {
        ImageCoord ans;
        ans.x = rand() % width;
        ans.y = rand() % height;
        return ans;
    }
    bool RandomBoolean() {
        return rand() % 2 == 1;
    }
};

static Random gRandom;

void GeneratorGaussKernel(int ksize, float sigma, float* kernel)
{
    int bufferSize = ksize * ksize;
    float sigmasigma2 = 2.0f * sigma * sigma;
    float sigmasigma2Inv = 1.f / sigmasigma2;
    float sigmasigma2PIInv = sigmasigma2Inv / 3.14159265358979f;
    int radius = ksize / 2;
    float sum = 0.f;
    for (int i = -radius; i <= radius; ++i) {
        for (int j = -radius; j <= radius; ++j) {
            kernel[(i + radius) * ksize + (j + radius)] = sigmasigma2PIInv * expf(-(i * i + j * j) * sigmasigma2Inv);
        }
    }

    for (int i = 0; i < bufferSize; ++i) {
        sum += kernel[i];
    }
    sum = 1.f / sum;
    for (int i = 0; i < bufferSize; ++i) {
        kernel[i] = kernel[i] * sum;
    }
}

void LeftAndRightMirrorImageUInt8(unsigned char* in, unsigned char* out, int width, int height)
{
    for (int i = 0; i < height; ++i) {
        int hoffset = i * width;
        for (int j = 0; j < width; ++j) {

            int woffset = (hoffset + j) * 3;
            int woffset_ = (hoffset + width - 1 - j) * 3;
            for (int n = 0; n < 3; ++n) {
                out[woffset_ + n] = in[woffset + n];
            }
        }
    }
}

void LeftAndRightMirrorImageFloat(float* in, float* out, int width, int height)
{
    for (int i = 0; i < height; ++i) {
        int hoffset = i * width;
        for (int j = 0; j < width; ++j) {
            int woffset = (hoffset + j) * 3;
            int woffset_ = (hoffset + width - 1 - j) * 3;
            for (int n = 0; n < 3; ++n) {
                out[woffset_ + n] = in[woffset + n];
            }
        }
    }
}

void UpAndDownMirrorImageFloat(float* in, float* out, int width, int height)
{
    int lineOffset = width * 3;
    int lineSize = lineOffset * sizeof(float);
    float* outTmp = out + lineOffset * height - lineOffset;
    float* inTmp = in;
    for (int i = 0; i < height; ++i) {
        memcpy_s(outTmp, lineSize, inTmp, lineSize);
        outTmp -= lineOffset;
        inTmp += lineOffset;
    }
}

void UpAndDownMirrorImageUInt8(unsigned char* in, unsigned char* out, int width, int height)
{
    int lineOffset = width * 3;
    int lineSize = lineOffset * sizeof(unsigned char);
    unsigned char* outTmp = out + lineOffset * height - lineOffset;
    unsigned char* inTmp = in;
    for (int i = 0; i < height; ++i) {
        memcpy_s(outTmp, lineSize, inTmp, lineSize);
        outTmp -= lineOffset;
        inTmp += lineOffset;
    }
}

#if 0

void Conv(float* in, float* out, int width, float* filter, int ksize) {

    int lineSize = width * 3;
    float* inTemp = in;
    float* outTemp = out;
    out[0] = 0.f; out[1] = 0.f; out[2] = 0.f;
    for (int i = 0; i < ksize; ++i) {
        for (int j = 0; j < ksize; ++j) {
            int xoffset = j * 3;
            out[0] += (*filter) * inTemp[xoffset + 0];
            out[1] += (*filter) * inTemp[xoffset + 1];
            out[2] += (*filter) * inTemp[xoffset + 2];
            filter++;
        }
        inTemp = inTemp + lineSize;
    }
}

void ImageFilterFloat(float* in, float* out, int width, int height, float* filterKernel, int kw, int kh)
{
    size_t size = (size_t)width * (size_t)height * sizeof(float) * 3;

    int startX = kw / 2;
    int endX = width - kw / 2;
    
    int startY = kh / 2;
    int endY = height - kh / 2;

    float* tempOut = out + (startY * width + startX) * 3;

    memset(out, 0, size);
    //memcpy_s(out, size, in, size);
    omp_set_num_threads(32);

#pragma omp parallel for
    for (int i = 0; i <= height - kh; ++i) {
        int yoffset = i * width * 3;
        for (int j = 0; j <= width - kw; ++j) {
            int xoffset = yoffset + j * 3;
            Conv((in + xoffset), (tempOut + xoffset), width, filterKernel, kw);
        }

    }
}
#elif 1

void Conv(float* in, float* out, int width, __m128* filter, int ksize) {

    int lineSize = width * 3;
    float* inTemp = in;
    float* outTemp = out;
    out[0] = 0.f; out[1] = 0.f; out[2] = 0.f;
    __m128 sum = _mm_set_ps1(0.f);
    for (int i = 0; i < ksize; ++i) {
        for (int j = 0; j < ksize; ++j) {
            int xoffset = j * 3;

            __m128 img_value = _mm_set_ps(1.f, inTemp[xoffset + 2], inTemp[xoffset + 1], inTemp[xoffset + 0]);

            sum = _mm_add_ps(_mm_mul_ps((*filter), img_value), sum);

            filter++;
        }
        inTemp = inTemp + lineSize;
    }
    out[0] = sum.m128_f32[0];
    out[1] = sum.m128_f32[1];
    out[2] = sum.m128_f32[2];
}

void ImageFilterFloat(float* in, float* out, int width, int height, float* filterKernel, int kw, int kh)
{
    size_t size = (size_t)width * (size_t)height * sizeof(float) * 3;

    int startX = kw / 2;
    int endX = width - kw / 2;

    int startY = kh / 2;
    int endY = height - kh / 2;

    float* tempOut = out + (startY * width + startX) * 3;

    memset(out, 0, size);

    __m128* filterKernel_m128 = (__m128*)_mm_malloc(kw * kh * sizeof(__m128), sizeof(__m128));
    for (int i = 0; i < kw * kh; ++i) {
        filterKernel_m128[i] = _mm_set_ps1(filterKernel[i]);
    }


    omp_set_num_threads(32);
#pragma omp parallel for
    for (int i = 0; i <= height - kh; ++i) {
        int yoffset = i * width * 3;
        for (int j = 0; j <= width - kw; ++j) {
            int xoffset = yoffset + j * 3;
            Conv((in + xoffset), (tempOut + xoffset), width, filterKernel_m128, kw);
        }

    }

    if (filterKernel_m128) {
        _mm_free(filterKernel_m128);
        filterKernel_m128 = NULL;
    }
}

#endif
void SaltAndPepperFloat(float* in, float* out, int width, int height, float minV, float maxV, float proportion)
{
    int coordNumber = (int)(width * height * proportion);

    if (in != out) {
        memcpy_s(out, width * height * 3 * sizeof(float), in, width * height * 3 * sizeof(float));
    }

    for (int i = 0; i < coordNumber; ++i) {
        ImageCoord coord = gRandom.RandomImageCoord(width, height);
        bool saltOrPepper = gRandom.RandomBoolean();
        float value = saltOrPepper ? minV : maxV;
        int x = coord.x;
        int y = coord.y;
        int offset = (y * width + x) * 3;
        for (int c = 0; c < 3; ++c) {
            out[offset + c] = value;
        }
    }
}

void SaltAndPepperUInt8(unsigned char* in, unsigned char* out, int width, int height, float minV, float maxV, float proportion)
{
    int coordNumber = (int)(width * height * proportion);

    if (in != out) {
        memcpy_s(out, width * height * 3 * sizeof(unsigned char), in, width * height * 3 * sizeof(unsigned char));
    }
    for (int i = 0; i < coordNumber; ++i) {
        ImageCoord coord = gRandom.RandomImageCoord(width, height);
        bool saltOrPepper = gRandom.RandomBoolean();
        float value = saltOrPepper ? minV : maxV;
        int x = coord.x;
        int y = coord.y;
        int offset = (y * width + x) * 3;
        for (int c = 0; c < 3; ++c) {
            out[offset + c] = (unsigned char)value;
        }
    }
}

void ImageMinMax(float* in, int width, int height, int channels, float* minV, float* maxV)
{
    float minValue = 99999.f;
    float maxValue = -minValue;
    int number = width * height * channels;
    for (int i = 0; i < number; ++i) {
        float value = in[i];
        if (value > maxValue) {
            maxValue = value;
        }
        if (value < minValue) {
            minValue = value;
        }
    }
    *minV = (float)minValue;
    *maxV = (float)maxValue;
}

void ImageMinMax(unsigned char* in, int width, int height, int channels, unsigned char* minV, unsigned char* maxV)
{
    int minValue = 256;
    int maxValue = -1;
    int number = width * height * channels;
    for (int i = 0; i < number; ++i) {
        int value = in[i];
        if (value > maxValue) {
            maxValue = value;
        }
        if (value < minValue) {
            minValue = value;
        }
    }
    *minV = (unsigned char)minValue;
    *maxV = (unsigned char)maxValue;
}

void ImageMulAAddBFloatFloat(float* in, float* out, int width, int height, int channels, float A, float B)
{
    int size = width * height * channels;
    for (int i = 0; i < size; ++i) {
        out[i] = in[i] * A + B;
    }
}

void ImageMulAAddBUInt8UInt8(unsigned char* in, unsigned char* out, int width, int height, int channels, float A, float B)
{
#define ALVACLAMP(x, minV, maxV) \
        (x) < (minV) ? (minV) : ((x) > (maxV) ? (maxV) : (x))
    int size = width * height * channels;
    for (int i = 0; i < size; ++i) {
        out[i] = (unsigned char)(ALVACLAMP(in[i] * A + B, 0, 255));
    }
#undef ALVACLAMP
}

void ImageMulAAddBUInt8Float(unsigned char* in, float* out, int width, int height, int channels, float A, float B)
{
    int size = width * height * channels;
    for (int i = 0; i < size; ++i) {
        out[i] = in[i] * A + B;
    }
}

void NormalizeUInt8Float(unsigned char* in, float* out, int width, int height, int channels, int type)
{
    unsigned char minV, maxV;
    ImageMinMax(in, width, height, channels, &minV, &maxV);
    int size = width * height * channels;
    float inv = 1.f / (maxV - minV);
    float offset = 0.f;
    if (type == 1) {
        inv *= 2.f;
        offset = -1.f;
    }
    for (int i = 0; i < size; ++i) {
        out[i] = (in[i] - minV) * inv + offset;
    }
}

void NormalizeFloatFloat(float* in, float* out, int width, int height, int channels, int type)
{
    float minV, maxV;
    ImageMinMax(in, width, height, channels, &minV, &maxV);
    int size = width * height * channels;
    float inv = 1.f / (maxV - minV);
    float offset = 0.f;
    if (type == 1) {
        inv *= 2.f;
        offset = -1.f;
    }
    for (int i = 0; i < size; ++i) {
        out[i] = (in[i] - minV) * inv + offset;
    }
}

void RGBAvgUInt8Float(unsigned char* in, float* out, int width, int height)
{
    int size = width * height;
    for (int i = 0; i < size; ++i) {
        float avg = (in[i * 3 + 0] + in[i * 3 + 1] + in[i * 3 + 2]) / 3.f;
        out[i * 3 + 0] = avg;
        out[i * 3 + 1] = avg;
        out[i * 3 + 2] = avg;
    }
}

void RGBAvgFloatFloat(float* in, float* out, int width, int height)
{
    int size = width * height;
    for (int i = 0; i < size; ++i) {
        float avg = (in[i * 3 + 0] + in[i * 3 + 1] + in[i * 3 + 2]) / 3.f;
        out[i * 3 + 0] = avg;
        out[i * 3 + 1] = avg;
        out[i * 3 + 2] = avg;
    }
}

static PyObject* GeneratorGaussKernel(PyObject* self, PyObject* args) {
    //int ksize, float sigma, float* kernel
    PyObject* pyobj_filter = NULL;
    int ksize;
    float sigma;
    int ret = PyArg_ParseTuple(args, "Oif", &pyobj_filter, &ksize, &sigma);

    PyArrayObject* oarr = (PyArrayObject*)pyobj_filter;

    float* data = (float*)(oarr->data);

    GeneratorGaussKernel(ksize, sigma, data);
    PyObject* result = PyUnicode_FromFormat("result:%s", "ok");
    return result;
}

static PyObject* LeftAndRightMirrorImageUInt8(PyObject* self, PyObject* args) {
    PyObject* pyobj_img = NULL;
    PyObject* pyobj_out_img = NULL;
    int width, height;
	int ret = PyArg_ParseTuple(args, "OOii", &pyobj_img, &pyobj_out_img, &width, &height);
    PyArrayObject* oarr = (PyArrayObject*)pyobj_img;

    PyArrayObject* oarr_out = (PyArrayObject*)pyobj_out_img;

    unsigned char* dataIn = (unsigned char*)(oarr->data);
    unsigned char* dataOut = (unsigned char*)(oarr_out->data);
    
    LeftAndRightMirrorImageUInt8(dataIn, dataOut, width, height);

    PyObject* result = PyUnicode_FromFormat("result:%s", "ok");
    return result;

    //std::cout << "";
}

static PyObject* LeftAndRightMirrorImageFloat(PyObject* self, PyObject* args) {

    PyObject* pyobj_img = NULL;
    PyObject* pyobj_out_img = NULL;
    int width, height;
    int ret = PyArg_ParseTuple(args, "OOii", &pyobj_img, &pyobj_out_img, &width, &height);

    PyArrayObject* oarr = (PyArrayObject*)pyobj_img;

    PyArrayObject* oarr_out = (PyArrayObject*)pyobj_out_img;

    float* dataIn = (float*)(oarr->data);    
    float* dataOut = (float*)(oarr_out->data);

    LeftAndRightMirrorImageFloat(dataIn, dataOut, width, height);

    PyObject* result = PyUnicode_FromFormat("result:%s", "ok");
    return result;
}

static PyObject* UpAndDownMirrorImageUInt8(PyObject* self, PyObject* args) {

    PyObject* pyobj_img = NULL;
    PyObject* pyobj_out_img = NULL;
    int width, height;
    int ret = PyArg_ParseTuple(args, "OOii", &pyobj_img, &pyobj_out_img, &width, &height);

    PyArrayObject* oarr = (PyArrayObject*)pyobj_img;

    PyArrayObject* oarr_out = (PyArrayObject*)pyobj_out_img;

    unsigned char* dataIn = (unsigned char*)(oarr->data);

    unsigned char* dataOut = (unsigned char*)(oarr_out->data);

    UpAndDownMirrorImageUInt8(dataIn, dataOut, width, height);
    PyObject* result = PyUnicode_FromFormat("result:%s", "ok");
    return result;
}

static PyObject* UpAndDownMirrorImageFloat(PyObject* self, PyObject* args) {

    PyObject* pyobj_img = NULL;
    PyObject* pyobj_out_img = NULL;
    int width, height;
    int ret = PyArg_ParseTuple(args, "OOii", &pyobj_img, &pyobj_out_img, &width, &height);

    PyArrayObject* oarr = (PyArrayObject*)pyobj_img;

    PyArrayObject* oarr_out = (PyArrayObject*)pyobj_out_img;

    float* dataIn = (float*)(oarr->data);
    float* dataOut = (float*)(oarr_out->data);

    UpAndDownMirrorImageFloat(dataIn, dataOut, width, height);
    PyObject* result = PyUnicode_FromFormat("result:%s", "ok");
    return result;
}

static PyObject* ImageFilterFloat(PyObject* self, PyObject* args) {
    //float* in, float* out, int width, int height, float* filterKernel, int kw, int kh
    PyObject* pyobj_img = NULL;
    PyObject* pyobj_out_img = NULL;
    PyObject* pyobj_filterKernel = NULL;
    int width, height;
    int kw, kh;
    int ret = PyArg_ParseTuple(args, "OOiiOii", &pyobj_img, &pyobj_out_img, &width, &height, &pyobj_filterKernel, &kw, &kh);

    PyArrayObject* oarr = (PyArrayObject*)pyobj_img;
    PyArrayObject* oarr_out = (PyArrayObject*)pyobj_out_img;
    PyArrayObject* kernel = (PyArrayObject*)pyobj_filterKernel;

    float* dataIn = (float*)(oarr->data);
    float* dataOut = (float*)(oarr_out->data);
    float* filter = (float*)(kernel->data);

    ImageFilterFloat(dataIn, dataOut, width, height, filter, kw, kh);
    PyObject* result = PyUnicode_FromFormat("result:%s", "ok");
    return result;
}

static PyObject* SaltAndPepperFloat(PyObject* self, PyObject* args) {
    //float* in, float* out, int width, int height, float minV, float maxV, float proportion
    PyObject* pyobj_img = NULL;
    PyObject* pyobj_out_img = NULL;
    int width, height;
    float minV, maxV, proportion;
    int ret = PyArg_ParseTuple(args, "OOiifff", &pyobj_img, &pyobj_out_img, &width, &height, &minV, &maxV, &proportion);

    PyArrayObject* oarr = (PyArrayObject*)pyobj_img;
    PyArrayObject* oarr_out = (PyArrayObject*)pyobj_out_img;

    float* dataIn = (float*)(oarr->data);
    float* dataOut = (float*)(oarr_out->data);

    SaltAndPepperFloat(dataIn, dataOut, width, height, minV, maxV, proportion);
    PyObject* result = PyUnicode_FromFormat("result:%s", "ok");
    return result;
}

static PyObject* SaltAndPepperUInt8(PyObject* self, PyObject* args) {
    //float* in, float* out, int width, int height, float minV, float maxV, float proportion
    PyObject* pyobj_img = NULL;
    PyObject* pyobj_out_img = NULL;
    int width, height;
    float minV, maxV, proportion;
    int ret = PyArg_ParseTuple(args, "OOiifff", &pyobj_img, &pyobj_out_img, &width, &height, &minV, &maxV, &proportion);

    PyArrayObject* oarr = (PyArrayObject*)pyobj_img;
    PyArrayObject* oarr_out = (PyArrayObject*)pyobj_out_img;

    unsigned char* dataIn = (unsigned char*)(oarr->data);
    unsigned char* dataOut = (unsigned char*)(oarr_out->data);

    SaltAndPepperUInt8(dataIn, dataOut, width, height, minV, maxV, proportion);
    PyObject* result = PyUnicode_FromFormat("result:%s", "ok");
    return result;
}

static PyObject* ImageMulAAddBFloatFloat(PyObject* self, PyObject* args) {
    //float* in, float* out, int width, int height, int channels, float A, float B
    PyObject* pyobj_img = NULL;
    PyObject* pyobj_out_img = NULL;
    int width, height, channels = 3;
    float A, B;
    int ret = PyArg_ParseTuple(args, "OOiiff", &pyobj_img, &pyobj_out_img, &width, &height, &A, &B);

    PyArrayObject* oarr = (PyArrayObject*)pyobj_img;
    PyArrayObject* oarr_out = (PyArrayObject*)pyobj_out_img;

    float* dataIn = (float*)(oarr->data);
    float* dataOut = (float*)(oarr_out->data);

    ImageMulAAddBFloatFloat(dataIn, dataOut, width, height, channels, A, B);
    PyObject* result = PyUnicode_FromFormat("result:%s", "ok");
    return result;
}

static PyObject* ImageMulAAddBUInt8Float(PyObject* self, PyObject* args) {
    //float* in, float* out, int width, int height, int channels, float A, float B
    PyObject* pyobj_img = NULL;
    PyObject* pyobj_out_img = NULL;
    int width, height, channels = 3;
    float A, B;
    int ret = PyArg_ParseTuple(args, "OOiiff", &pyobj_img, &pyobj_out_img, &width, &height, &A, &B);

    PyArrayObject* oarr = (PyArrayObject*)pyobj_img;
    PyArrayObject* oarr_out = (PyArrayObject*)pyobj_out_img;

    unsigned char* dataIn = (unsigned char*)(oarr->data);
    float* dataOut = (float*)(oarr_out->data);

    ImageMulAAddBUInt8Float(dataIn, dataOut, width, height, channels, A, B);
    PyObject* result = PyUnicode_FromFormat("result:%s", "ok");
    return result;
}

static PyObject* ImageMulAAddBUInt8UInt8(PyObject* self, PyObject* args) {
    //float* in, float* out, int width, int height, int channels, float A, float B
    PyObject* pyobj_img = NULL;
    PyObject* pyobj_out_img = NULL;
    int width, height, channels = 3;
    float A, B;
    int ret = PyArg_ParseTuple(args, "OOiiff", &pyobj_img, &pyobj_out_img, &width, &height, &A, &B);

    PyArrayObject* oarr = (PyArrayObject*)pyobj_img;
    PyArrayObject* oarr_out = (PyArrayObject*)pyobj_out_img;

    unsigned char* dataIn = (unsigned char*)(oarr->data);
    unsigned char* dataOut = (unsigned char*)(oarr_out->data);

    ImageMulAAddBUInt8UInt8(dataIn, dataOut, width, height, channels, A, B);
    PyObject* result = PyUnicode_FromFormat("result:%s", "ok");
    return result;
}

static PyObject* NormalizeUInt8Float(PyObject* self, PyObject* args) {
    // unsigned char* in, float* out, int width, int height, int channels, int type
    PyObject* pyobj_img = NULL;
    PyObject* pyobj_out_img = NULL;
    int width, height, channels = 3;
    int type;
    int ret = PyArg_ParseTuple(args, "OOiii", &pyobj_img, &pyobj_out_img, &width, &height, &type);

    PyArrayObject* oarr = (PyArrayObject*)pyobj_img;
    PyArrayObject* oarr_out = (PyArrayObject*)pyobj_out_img;

    unsigned char* dataIn = (unsigned char*)(oarr->data);
    float* dataOut = (float*)(oarr_out->data);

    NormalizeUInt8Float(dataIn, dataOut, width, height, channels, type);
    PyObject* result = PyUnicode_FromFormat("result:%s", "ok");
    return result;
}

static PyObject* NormalizeFloatFloat(PyObject* self, PyObject* args) {
    // unsigned char* in, float* out, int width, int height, int channels, int type
    PyObject* pyobj_img = NULL;
    PyObject* pyobj_out_img = NULL;
    int width, height, channels = 3;
    int type;
    int ret = PyArg_ParseTuple(args, "OOiii", &pyobj_img, &pyobj_out_img, &width, &height, &type);

    PyArrayObject* oarr = (PyArrayObject*)pyobj_img;
    PyArrayObject* oarr_out = (PyArrayObject*)pyobj_out_img;

    float* dataIn = (float*)(oarr->data);
    float* dataOut = (float*)(oarr_out->data);

    NormalizeFloatFloat(dataIn, dataOut, width, height, channels, type);
    PyObject* result = PyUnicode_FromFormat("result:%s", "ok");
    return result;
}

static PyObject* RGBAvgUInt8Float(PyObject* self, PyObject* args) {
    // unsigned char* in, float* out, int width, int height
    PyObject* pyobj_img = NULL;
    PyObject* pyobj_out_img = NULL;
    int width, height;
    int ret = PyArg_ParseTuple(args, "OOii", &pyobj_img, &pyobj_out_img, &width, &height);

    PyArrayObject* oarr = (PyArrayObject*)pyobj_img;
    PyArrayObject* oarr_out = (PyArrayObject*)pyobj_out_img;

    unsigned char* dataIn = (unsigned char*)(oarr->data);
    float* dataOut = (float*)(oarr_out->data);

    RGBAvgUInt8Float(dataIn, dataOut, width, height);
    PyObject* result = PyUnicode_FromFormat("result:%s", "ok");
    return result;
}

static PyObject* RGBAvgFloatFloat(PyObject* self, PyObject* args) {
    // unsigned char* in, float* out, int width, int height
    PyObject* pyobj_img = NULL;
    PyObject* pyobj_out_img = NULL;
    int width, height;
    int ret = PyArg_ParseTuple(args, "OOii", &pyobj_img, &pyobj_out_img, &width, &height);

    PyArrayObject* oarr = (PyArrayObject*)pyobj_img;
    PyArrayObject* oarr_out = (PyArrayObject*)pyobj_out_img;

    float* dataIn = (float*)(oarr->data);
    float* dataOut = (float*)(oarr_out->data);

    RGBAvgFloatFloat(dataIn, dataOut, width, height);
    PyObject* result = PyUnicode_FromFormat("result:%s", "ok");
    return result;
}

static PyMethodDef DemoMethods[] = {
    {"LeftAndRightMirrorImageUInt8", (PyCFunction)LeftAndRightMirrorImageUInt8, METH_VARARGS | METH_KEYWORDS, "I guess here is description." },
    {"LeftAndRightMirrorImageFloat", (PyCFunction)LeftAndRightMirrorImageFloat, METH_VARARGS | METH_KEYWORDS, "I guess here is description." },
    {"UpAndDownMirrorImageUInt8", (PyCFunction)UpAndDownMirrorImageUInt8, METH_VARARGS | METH_KEYWORDS, "I guess here is description." },
    {"UpAndDownMirrorImageFloat", (PyCFunction)UpAndDownMirrorImageFloat, METH_VARARGS | METH_KEYWORDS, "I guess here is description." },
    {"ImageFilterFloat", (PyCFunction)ImageFilterFloat, METH_VARARGS | METH_KEYWORDS, "I guess here is description." },
    {"SaltAndPepperFloat", (PyCFunction)SaltAndPepperFloat, METH_VARARGS | METH_KEYWORDS, "I guess here is description." },
    {"SaltAndPepperUInt8", (PyCFunction)SaltAndPepperUInt8, METH_VARARGS | METH_KEYWORDS, "I guess here is description." },
    {"ImageMulAAddBFloatFloat", (PyCFunction)ImageMulAAddBFloatFloat, METH_VARARGS | METH_KEYWORDS, "I guess here is description." },
    {"ImageMulAAddBUInt8Float", (PyCFunction)ImageMulAAddBUInt8Float, METH_VARARGS | METH_KEYWORDS, "I guess here is description." },
    {"ImageMulAAddBUInt8UInt8", (PyCFunction)ImageMulAAddBUInt8UInt8, METH_VARARGS | METH_KEYWORDS, "I guess here is description." },
    {"NormalizeUInt8Float", (PyCFunction)NormalizeUInt8Float, METH_VARARGS | METH_KEYWORDS, "I guess here is description." },
    {"NormalizeFloatFloat", (PyCFunction)NormalizeFloatFloat, METH_VARARGS | METH_KEYWORDS, "I guess here is description." },
    {"RGBAvgUInt8Float", (PyCFunction)RGBAvgUInt8Float, METH_VARARGS | METH_KEYWORDS, "I guess here is description." },
    {"RGBAvgFloatFloat", (PyCFunction)RGBAvgFloatFloat, METH_VARARGS | METH_KEYWORDS, "I guess here is description." },
    {"GeneratorGaussKernel", (PyCFunction)GeneratorGaussKernel, METH_VARARGS | METH_KEYWORDS, "I guess here is description." },
    {NULL, NULL, 0, NULL}
};

static struct PyModuleDef demoModule = {
    PyModuleDef_HEAD_INIT,
    "mirror",
    NULL,
    -1,
    DemoMethods
};

PyMODINIT_FUNC
PyInit_ImgProcessing(void) {  // ImgProcessing 为生成的dll名称
    return PyModule_Create(&demoModule);
}

Python 代码

把C++ 编译出的ImgProcessing.dll 改为 ImgProcessing.pyd 并 拷贝到Python工程下

# -*- coding:utf-8 -*-
import cv2
import random
import numpy as np
import ImgProcessing as aug


class AugmentImagesBase:
    def _gauss(self, x, y, sigma=1.):
        Z = 2 * np.pi * sigma ** 2
        kernel_value = 1 / Z * np.exp(-(x ** 2 + y ** 2) / 2 / sigma ** 2)
        return kernel_value

    def _gauss_kernel(self, kwidth, kheight, kchannel=1):
        kernels = np.zeros((kheight, kwidth, kchannel, 1), np.float32)
        mid = np.floor(kwidth / 2)
        for kernel_idx in range(kchannel):
            for i in range(kheight):
                for j in range(kwidth):
                    kernels[i, j, kernel_idx, 0] = self._gauss(i - mid, j - mid)
        if kchannel == 1:
            kernels = np.reshape(kernels, (kheight, kwidth))
        return kernels

    def left_right_flip(self, img_in, img_out, width=336, height=192):
        aug.AlvaLeftAndRightMirrorImageUInt8(img_in, img_out, width, height)
        return img_out

    def up_down_flip(self, img_in, img_out, width=336, height=192):
        aug.AlvaUpAndDownMirrorImageUInt8(img_in, img_out, width, height)
        return img_out

    def filtering(self, img_in, img_out, width=336, height=192, kernel=None, kwidth=3, kheight=3):
        aug.AlvaImageFilterFloat(img_in, img_out, width, height, kernel, kwidth, kheight)
        return img_out

    def pepper_salt(self, img_in, img_out, width=336, height=192, min_v=0, max_v=255, proportion=0.1):
        rand_proportion = random.uniform(0., proportion)
        aug.AlvaSaltAndPepperUInt8(img_in, img_out, width, height, min_v, max_v, rand_proportion)
        return img_out

    def contrast(self, img_in, img_out, width=336, height=192, a=0.6, b=0.4):
        aug.AlvaImageMulAAddBUInt8UInt8(img_in, img_out, width, height, a, b)
        return img_out

    def average_rgb(self, img_in, img_out, width=336, height=192):
        img_in = img_in.astype(np.float32)
        img_out = img_out.astype(np.float32)
        aug.AlvaRGBAvgFloatFloat(img_in, img_out, width, height)
        img_out = img_out.astype(np.uint8)
        return img_out

    def normalize(self, img_in, img_out, width=336, height=192, type=1):
        aug.AlvaNormalizeUInt8Float(img_in, img_out, width, height, type)
        return img_out

    def normal(self, img_in, img_out):
        return img_in

    def rota_180(self, img_in, img_out):
        return cv2.rotate(img_in, cv2.ROTATE_180)

    def rand_aug(self, img_in):
        img_in = np.asarray(img_in, dtype=np.uint8)
        img_out = np.ones_like(img_in).astype(np.uint8)
        aug_func = {
            "left_right_flip": self.left_right_flip,
            'up_down_flip': self.up_down_flip,
            'pepper_salt': self.pepper_salt,
            'contrast': self.contrast,
            'average_rgb': self.average_rgb,
            "normal": self.normal,
            "rota_180": self.rota_180,
        }
        img_out_curr = np.ones_like(img_in[0]).astype(np.uint8)
        aug_names = []
        for i in range(img_in.shape[0]):
            aug_name = random.sample(list(aug_func.keys()), 1)[0]
            img_out_curr = aug_func[aug_name](np.squeeze(img_in[i]), img_out_curr)
            img_out[i] = img_out_curr
            aug_names.append(aug_name)
        return img_out, aug_names


def image_aug(img_path):
    import cv2
    import time
    aug_tools = AugmentImagesBase()
    kernel = aug_tools._gauss_kernel(5, 5)
    img_in = cv2.imread(img_path).astype(np.float32)
    img_out = np.ones_like(img_in).astype(np.float32)
    time1 = time.time()
    for i in range(1000):
        # img_out, aug_names = aug_tools.average_rgb(img_in, img_out)
        img_out = aug_tools.filtering(img_in, img_out, kernel=kernel, kwidth=5, kheight=5)

    time2 = time.time()
    print("end time:", time2 - time1)
    # cv2.imshow(aug_names[0], img_out[0])
    cv2.imshow("aug img", img_out.astype(np.uint8))
    cv2.imshow('src img', img_in.astype(np.uint8))
    cv2.waitKey(0)


if __name__ == "__main__":
    img_path = r"G:\20210917\img\1.jpg"
    image_aug(img_path)

PyArg_ParseTuple 的使用

见：PyArg_ParseTuple

darkerJ
关注
  • 0
    点赞
  • 踩
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
Python Numpy库的C ++实现-C/C++开发
05-26
NumCpp：Python NumPy库的仅模板化C ++实现作者：David Pilger [受电子邮件保护]版本：1.3许可证版权2020 David Pilger特此授权NumCpp：Python NumPy库的仅模板化C ++实现作者：David Pilger [版本：1.3许可版权所有2020 David Pilger特此免费授予任何获得本软件和相关文档文件（“软件”）副本的任何人，以不受限制地（包括但不限于）处理本软件限制使用，复制，修改，合并，发布，分发，再许可和/或出售So副本的权利
在 C++ 代码中调用 NumPy
程序猿视角
05-22 1800
要在 C++ 代码中调用 NumPy，可以使用 Boost.Python 库。为了这样做，需要安装 Boost 开发库和 NumPy ，并使用 BOOST_PYTHON_MODULE 宏定义来创建一个 Python 模块。以下是一个简单的示例，演示如何创建一个名为。
numpy库的基础知识介绍与基本使用
m0_61959040的博客
04-16 1339
1.数据的维度 维度：一组数据的组织形式 一维数据由对等关系的有序或无序数据构成，采用线性方式组织。 可以用列表，数组（python无但是nupmy库中有）表示一维数据（区别列表，数组中数据可以是不同类型的） 二维数据由多个一维数据构成，是一维数据的组合形式 可以用列表类型。 多维数据有一维或二维数据在新维度上扩展形成 可以用列表类型。 高维数据，仅使用最基本的二元关系展示数据间的复杂结构 字典类型或数据表示格式。 2.numpy的数组对象ndarray numpy是一
C++和Python混合编程：C++版的Numpy（Xtensor.hpp）安装和使用
养乐多的博客
10-14 2285
xtensor是一个 C++ 库，用于使用多维数组表达式进行数值分析。c++ 的 xtensor 库对标 Python 的 Numpy 库，Numpy 在 Python 中使用很方便，但是在 C++ 中想要使用这么方便的工具就需要 stensor。xtensor 支持延迟广播。本文介绍 xtensor.hpp 在 C++ 中的安装使用。
NumCpp(C++版本Numpy)使用笔记
热门推荐
Tosonw的博客
06-13 2万+
一、简介 NumCpp：Python NumPy库的一个Templatized Header Only C ++实现 NumCpp 是一个高性能的数学计算 C++ 库，它提供了一个简单的 Numpy/Matlab 类似的接口。 地址 https://github.com/dpilger26/NumCpp 文档地址 https://dpilger26.github.io/NumCpp/doxygen...
NumPy库的介绍与使用教程(一）
☆下山☆的博客
07-01 2718
       “实践是检验真理的唯一标准”。对我个人来讲，学习任何一门python语言的库最好的方法就是实践。所以我希望当你在读我这篇文章的时候能够拿出你的电脑，一边阅读一边实践。 NumPy库简介        NumPy(Numerical Python) 是 Python 语言的一个扩展程序库。其中提供了许多向量和矩阵操作，能让用户轻松完成最优化、线性代数、积分、
C++调用Python并传参
03-13
C#调用C++,C++调用Python并传参，包括C++向python传递图片数据，C++向python传递组数据，C++向python传递字符串数据，C++向python传递结构体数据，C++初始化python类数据并调用方法，实际案例，能直接运行，需要安装...
本资源为python和C++联合编程的示例vs工程源码，为g2o的ba优化程序 供参考
最新发布
11-18
我们将使用 Python 3.7 和 C++，并利用 numpy 数组格式接收 Python 侧的两幅图像，并利用 g2o 进行 BA 优化，最后将优化结果以 numpy 数组形式返回 Python。对应博文《C++加持让python程序插上翅膀——利用pybind11...
python实现C++程序g2o的ba优化模块调用，python侧的程序
11-18
我们将使用 Python 3.7 和 C++，并利用 numpy 数组格式接收 Python 侧的两幅图像，并利用 g2o 进行 BA 优化，最后将优化结果以 numpy 数组形式返回 Python。博文《C++加持让python程序插上翅膀——利用pybind11进行...
python调用c
01-29
文档中python调用c或c++的详细过程，这是以前的一个项目需要，当时用到numpy。大家需要的可以下载。
VS2019C++调用Python方法绘制频率直方图和正态分布
05-11
在VS中C++调用python.py文件，通过import numpy, matlab,scripy,进行的库功能进行绘制频率直方图和正态分布曲线； 博文链接： https://mp.csdn.net/mp_blog/creation/editor/124712486
C++版NumPy-Eigen库快速入门
ISmileLi的博客
12-19 4297
Eigen库的使用零、前言一、Eigen矩阵类1、矩阵类模板参数2、向量3、特值动态4、矩阵和向量的初始化5、矩阵和向量的存取6、逗号初始化7、调整大小二、矩阵和向量算法1、加减2、标量乘法和除法3、转置、共轭和伴随4、矩阵矩阵和矩阵向量乘法5、点积和叉积6、基本算术归约运算7、操作的有效性三、数组类和按系数运算1、数组类2、访问数组内的值3、加减4、数组乘法5、其他系数运算6、在数组和矩阵表达式之间转换四、块操作1、使用块操作2、列和行3、与角相关的操作4、向量的块运算五、Eigen高级初始化1、逗号初始
Numpy库的介绍及使用
qq_53763141的博客
12-20 9187
本文章介绍了python常用库numpy的常用方法，对于数据分析入门、机器学习的入门有良好的帮助。
Python调用c/c++动态库（一）
ANTI4T的博客
03-24 167
Python调用c/c++动态库（一）_python调用c++动态库_giveaname的博客-CSDN博客
C++ 扩展python(四）传递numpy（使用SWIG）
DS00HY的小博客
03-20 1048
前期准备： numpy.i文件，有些conda下载的numpy库里面自带，而有些没有（比如我的，，），可以去下载numpy.i源码。 我们实现一个cos_doubles模块的cos_double函数来对numpy数组求cos, 并返回他的cos值到一个新的numpy数组中： 测试样例test.py如下： import numpy as np import cos_doubles x =...
Python3：PyArg_ParseTuple与Py_BuildValue
风静如云的博客
08-13 1638
需要注意的是：使用PyArg_ParseTuple获取一个来自于Python传递的字符串类型(char *)的参数时，传递的是一个指向指针的指针，也就是说PyArg_ParseTuple会修改s所指向的位置，该位置上会保存Python传递的字符串，并且该位置由Python维护，C函数不能主动将其释放。需要注意的是：使用Py_BuildValue构建返回值时，如果C函数通过malloc分配了内存空间并用于构建Py_BuildValue，那么当构建完成后，需要C函数调用free释放内存空间。......
python如何调用c++numpy.ndarray代码_python numpy ndarray.ctypes用法及代码示例
weixin_34342091的博客
01-14 298
一个用于简化数组与ctypes模块的交互的对象。此属性创建一个对象，使使用ctypes模块调用共享库时更易于使用数组。返回的对象除其他外，还具有数据，形状和步幅属性(请参阅下面的注释)，它们本身返回ctypes对象，这些对象可用作共享库的参数。参数：None：返回值：c： ： Python对象拥有属性数据，形状，步幅等注意：以下是在“Guide to NumPy”中记录的该对象的公共属性(我们省略...
python数组数据类型转换_python – 将PyArrayObject数据类型转换为C数组
weixin_39990410的博客
11-29 1004
我想在C扩展中使用我的Numpy数组.在这种情况下,许多示例使用PyArrayObject的结构,array-> data,array-> strides [0],array-> strides [1],…如果我想以更熟悉(或更整洁)的方式到达我的数组,以及像阵列[i] [j]我该怎么办呢？我应该转换(bool *)array->数据并使用我创建的C数组吗？ (我的元素是bools)我现在的功能声...
Python调用C++代码用法——Linux
01-02 1976
在项目开发中，有时会使用到多种编程语言，比如部分功能是C/C++代码实现的，而另一部分是Python代码实现的，这样就可能需要使用多种编程语言。当然，也可以把C/C++代码转成Python，但这样可能费时；也有可能某个模块用Python来实现，速度很慢，但用C/C++速度却很快，这就会使用到用Python来调用C/C++代码。本文主要说明如何使用Python来调用C/C++代码，且是基于Linux平台，Linux上编译生成动态共享库so文件，然后Python使用ctype模块来调用so文件里的函数。
python调用C++函数传入ndarray数据
06-07
在Python中调用C++函数并传递ndarray数据，可以使用NumPy库将ndarray数据转换为C++中的数组类型。下面是一个示例代码，展示了如何在Python中调用一个接受ndarray数据的C++函数： 假设我们有一个C++函数，它接受一个浮点型数组类型`float*`和数组的长度`int`： ```c++ #include <iostream> void my_function(float* data, int length) { for (int i = 0; i < length; i++) { std::cout << data[i] << " "; } std::cout << std::endl; } ``` 现在，我们可以在Python中使用NumPy库来创建一个ndarray数据，并将其传递给这个C++函数。下面是示例代码： ```python import numpy as np import ctypes # 加载C++编译后的动态链接库 lib = ctypes.cdll.LoadLibrary('./libexample.so') # 定义函数参数类型 lib.my_function.argtypes = [np.ctypeslib.ndpointer(dtype=np.float32, ndim=1), ctypes.c_int] # 创建一个ndarray数据，并将其传递给C++函数 data = np.array([1.0, 2.0, 3.0], dtype=np.float32) lib.my_function(data, len(data)) ``` 在上面的示例代码中，我们首先使用`cdll.LoadLibrary()`函数加载C++编译后的动态链接库。接着，我们使用`np.ctypeslib.ndpointer()`函数定义了C++函数的参数类型，其中`dtype`参数表示数组中元素的数据类型，`ndim`参数表示数组的维度。然后，我们将创建的ndarray数据`data`和数组的长度传递给C++函数`my_function()`。 需要注意的是，在传递ndarray数据时，需要正确地设置数据类型和维度。如果数据类型和维度设置不正确，可能会导致程序出错或结果不正确。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值