我正在進行一些優化，并希望將某些部分從 python 轉換為 c

是否可以將此代碼轉換為 c opencv？

python代碼使用numpy

import numpy as np
from PIL import Image

pil_img = Image.open(input_filename)
img = np.array(pil_img)
pixels = img.reshape((-1, 3))

num_pixels = pixels.shape[0]
num_samples = int(num_pixels*5)

idx = np.arange(num_pixels)
np.random.shuffle(idx)

samples = pixels[idx[:num_samples]]

更新

std::vector<uchar> sample_pixels(const cv::Mat& m, int sample_percent=5){
    assert(m.isContinuous());
    
    const auto* input = m.ptr<uchar>();
    
    int
        num_pixels      = m.total(),
        num_samples     = num_pixels * sample_percent;
    
    std::cout
        << "num pixels:  " << num_pixels << '\n'
        << "num samples: " << num_samples << '\n';
    
    std::vector<uchar> samples(num_samples);
    
    //  Fills idx with sequentially increasing values
    std::vector<int> idx(num_pixels);
    std::iota(idx.begin(), idx.end(), 0);
    
    //  Shuffle idx
    std::mt19937 engine(0);
    std::shuffle(idx.begin(), idx.end(), engine);
    
    for(int i = 0; i < num_samples; i  ){
        //samples[i] = input[idx[i]];
    }
    
    //auto output_mat = cv::Mat(samples, false);
    //cv::imwrite("enhance-samples.png", output_mat);
    
    return samples;
}

uj5u.com熱心網友回復：

這是 C 11 中的等效代碼。這應該比你的 python 代碼快幾倍。

#include <random>
#include <numeric>

#include <opencv2/opencv.hpp>

void shuffling(const std::string &input_filename, const std::string &output_filename) {

    // ========== UPDATE ==========
    const cv::Mat plain_input_mat = cv::imread(input_filename, -1);
    // Equivalent to img.reshape((-1, 3))
    const cv::Mat input_mat = plain_input_mat.reshape(3);
    // ============================

    // By doing this, you can access the pixels without any extra checks.
    assert(input_mat.isContinuous());
    const auto *input = input_mat.ptr<cv::Vec3b>();

    const auto num_samples = input_mat.total();
    std::vector<cv::Vec3b> output(num_samples);

    std::vector<int> idx(input_mat.total());
    std::iota(idx.begin(), idx.end(), 0);  // Equivalent to arange.

    // Note: numpy uses PCG64 which does not exist in the std library.
    std::mt19937 engine(0);
    std::shuffle(idx.begin(), idx.end(), engine);

    for (int i = 0; i < num_samples; i  ) {
        output[i] = input[idx[i]];
    }

    // Save as an image if necessary.
    auto output_mat = cv::Mat(output, false);
    cv::imwrite(output_filename, output_mat);
}

還有一些附加說明。

注1：由于python和std的shuffle演算法不同，結果并不完全相同。

注意2：使用您的代碼，num_samples不能大于輸入影像中的像素數，這似乎是一個錯誤。請檢查samples.

注意 3：在這兩種實作中，最昂貴的部分是shuffle. Python 的 60% 和 C 的 80% 以上都花在了這里。如果您想進一步優化，這絕對是您應該利用的地方。

轉載請註明出處，本文鏈接：https://www.uj5u.com/houduan/471838.html

標籤：Python C 麻木的 opencv

上一篇：錯誤：(-215:Assertionfailed)(int)_numAxes==inputs[0].size()infunction'getMemoryShapes'

下一篇：如何使用hsv顏色模型和opencv檢測影像的確切顏色

使用numpy將python轉換為使用opencv的c

更新