Python版OpenCVでの行列操作をC++版で行いたい

Question

Python版OpenCVの機能をC++に移植しているのですが、以下のPythonコードをC++版のOpenCVでどのように書けば良いのかわからなくなりました。一行で書けなくても結構ですので、どなたか助けて頂けると幸いです。
imgA,B,Cはそれぞれfloat型1チャンネル画像(numpyの2darray)です。

imgA[imgB > 0] = imgC[imgB > 0]

Comments

[stdio]

imgA[imgB > 0] = imgC[imgC > 0]
とはC++の計算式では、
imgA[0or1] = imgC[0or1];
となるので、imgのA,B,Cを配列だと仮定するならかなり難しい事になりますね。
まず、imgのA,B,Cは配列なのか、リストなのか文面から分かりません。この辺を定義して下さるとアドバイスしやすくなりますよ。

[tiitoi]

C++ に移植する以前に numpy でもこのような演算はできないはずですよ。
imgA.shape = imgB.shape = imgC.shape で
imgA[imgB > 0] = imgC[imgB > 0]
ならわかりますが、
imgA[imgB > 0] = imgC[imgC > 0]
だと、imgB > 0 が True の要素と imgC > 0 が True の要素が一致するとは限らないのでこのような代入はできません。

[stdio]

imgC[imgC > 0]は、
まぁ、ただの誤字じゃない？

[maskey]

ご指摘の通りimgC[imgB>0]の誤りでした。修正致しました。

[yominet]

最初
imgA[(imgB>0)&(imgC>0)] = imgC[(imgB>0)&(imgC>0)]
こうしたいのかと思いました

Answer 1

どうやら配列のようですのでお答えいたします。
結論から言ってその記載方法では出来ません。
この様にしてみてはいかがでしょうか？

void img(int count1,int count2){
    if(count1 < 0) count1 = 0;
    if(count2 < 0) count2 = 0;
    imgA[count1] = imgC[count2];
}

img(imgB.length,imgB.length);

Answer 2

cv::Mat::copyTo() をお使いください。

void cv::Mat::copyTo(OutputArray m, InputArray mask) const

• m: Destination matrix. If it does not have a proper size or type before the operation, it is reallocated.
• mask: Operation mask of the same size as *this. Its non-zero elements indicate which matrix elements need to be copied. The mask has to be of type CV_8U and can have 1 or multiple channels.

#include <iostream>

#include <opencv2/opencv.hpp>


int main()
{
    // mat1[mat3 > 0] = mat2[mat3 > 0] と以下は同じ
    
    cv::Mat mat1 = (cv::Mat_<float>(3, 3) << 1, 2, 3, 4, 5, 6, 7, 8, 9);
    std::cout << mat1 << std::endl;
    //    [1, 2, 3;
    //     4, 5, 6;
    //     7, 8, 9]

    cv::Mat mat2 =
        (cv::Mat_<float>(3, 3) << 11, 12, 13, 14, 15, 16, 17, 18, 19);
    std::cout << mat2 << std::endl;
    //    [11, 12, 13;
    //     14, 15, 16;
    //     17, 18, 19]

    cv::Mat mat3 = (cv::Mat_<float>(3, 3) << -1, -1, 1, -1, -1, -1, 1, -1, 1);
    std::cout << mat3 << std::endl;
    //    [-1, -1, 1;
    //     -1, -1, -1;
    //     1, -1, 1]

    // mat3 > 0 の要素のみ mat2 の値を mat1 にコピーする。
    mat2.copyTo(mat1, mat3 > 0);
    std::cout << mat1 << std::endl;
    //    [1, 2, 13;
    //     4, 5, 6;
    //     17, 8, 19]
}

Comments

[maskey]

ありがとうございます。望んだ様に動きました。