Opencv obatin 특정 픽셀 마스크를 기반으로 한 RGB 값

Question

제목이 명확하지 않을 수 있지만 다음 설명을주의 깊게 살펴보십시오. 미리 감사드립니다.

은 내가 RGB 이미지와 바이너리 마스크 이미지가 있습니다

Mat img = imread("test.jpg") 
Mat mask = Mat::zeros(img.rows, img.cols, CV_8U); 

은, 마스크에 어떤 사람을 지정 사람의 수는이 좌표를 기반으로, N. 이제 제로가 아닌 좌표를 알고 가정을, 우리 확실히 원래 어 이미지의 대응하는 픽셀의 RGB 값을 얻을 수있는이 다음과 같은 코드에 의해 수행 될 수 알고

Mat colors = Mat::zeros(N, 3, CV_8U); 
int counter = 0; 
for (int i = 0; i < mask.rows; i++) 
{ 
    for (int j = 0; j < mask.cols; j++) 
    { 
     if (mask.at<uchar>(i, j) == 1) 
     { 
      colors.at<uchar>(counter, 0) = img.at<Vec3b>(i, j)[0]; 
      colors.at<uchar>(counter, 1) = img.at<Vec3b>(i, j)[1]; 
      colors.at<uchar>(counter, 2) = img.at<Vec3b>(i, j)[2]; 
      counter++; 
     } 
    } 
} 

다음과 같은 좌표가 될 것이다 enter image description here

,617,

그러나 for 루프의이 두 계층에는 너무 많은 시간이 걸립니다. 색상을 알아내는 더 빠른 방법이 있는지 궁금 해서요. 사람들이 제가 전달하려고 한 것을 이해할 수 있기를 바랍니다.

PS : 나는 파이썬을 사용할 수 있다면, 이것은 단지 한 문장으로 수행 할 수 있습니다 : 당신이 C++를 사용하는 다른 오픈 소스 라이브러리를 사용하여 유연 경우, Armadillo을 시도

colors = img[mask == 1] 

Answer 1

. 모든 선형 대수 연산을 수행 할 수 있으며 위 코드를 한 줄로 줄일 수도 있습니다 (파이썬 코드 조각과 유사 함).

또는

는 findNonZero() 기능을 사용해보십시오 0이 아닌 값을 포함하는 이미지의 모든 좌표를 찾을 수 있습니다. 이것을 확인하십시오 : https://stackoverflow.com/a/19244484/7514664

Answer 2

.at() 방법은 C++에서 Mat 값에 액세스하는 가장 느린 방법입니다. 포인터를 사용하는 것이 가장 빠르지 만 모범 사례는 반복자입니다. OpenCV tutorial on scanning images을 참조하십시오.

파이썬 구문이 이런 식으로 멋지더라도, 하루가 끝날 때 모든 요소를 ​​반복해야합니다. 그리고이 전에 약간의 오버 헤드가 있기 때문에, 포인터로 C++ 루프보다 사실상 느립니다. 라이브러리에 관계없이 모든 요소를 ​​반복해야 할 필요가 있습니다. 모든 요소에 대한 마스크와 비교할 수 있습니다.

Answer 3

컴파일 최적화가 활성화와 함께,이 버전을 프로파일 링하려고하고 빠른 경우 우리에게 : 당신이 색상의 크기를 알고있는 경우

vector<Vec3b> colors; 
if (img.isContinuous() && mask.isContinuous()) { 
    auto pimg = img.ptr<Vec3b>(); 
    for (auto pmask = mask.datastart; pmask < mask.dataend; ++pmask, ++pimg) { 
     if (*pmask) 
      colors.emplace_back(*pimg); 
    } 
} 
else { 
    for (int r = 0; r < img.rows; ++r) { 
     auto prowimg = img.ptr<Vec3b>(r); 
     auto prowmask = img.ptr(r); 
     for (int c = 0; c < img.cols; ++c) { 
      if (prowmask[c]) 
       colors.emplace_back(prowimg[c]); 
     } 
    } 
} 

는, 미리위한 공간을 보유하고 있습니다.