OpenCV - 从BGRA输出创建CV_8UC3(3通道)Mat

时间:2016-06-05 12:04:29

标签: c++ opencv format pixel

我有一个AVCaptureSession输出CMSampleBuffer像素BGRA。我尝试使用数据指针以最有效的方式仅从Mat创建BGR个对象。

 CVPixelBufferLockBaseAddress(imageBuffer,0);
 void *baseaddress = CVPixelBufferGetBaseAddress(imageBuffer);
 CGRect videoRect = CGRectMake(0.0f, 0.0f, CVPixelBufferGetWidth(imageBuffer), CVPixelBufferGetHeight(imageBuffer));
 size_t bytesPerRow = (((CVPixelBufferGetBytesPerRow(imageBuffer)
 cv::Mat frame(videoRect.size.height, videoRect.size.width,CV_8UC3  , baseaddress,    bytesPerRow);

但它似乎没有用。有什么建议?

1 个答案:

答案 0 :(得分:2)

Mat中的数据必须是连续的(最后在每行末尾有一些额外的填充)。

您的数据如下:

row 0: B G R A B G R A ... B G R A <padding>
row 1: B G R A B G R A ... B G R A <padding>
...

无法读取以下数据:

row 0: B G R - B G R - ... B G R - <padding>
row 1: B G R - B G R - ... B G R - <padding>
...

您可以做的是将您的BGRA数据包装在CV_8UC4矩阵中(即在您的数据上构建矩阵标头):

cv::Mat frameBGRA (height, width, CV_8UC4, buffer);

然后复制仅显示您需要的字节:

cv::Mat frameBGR;
cv::cvtColor(frameBGRA, frameBGR, cv::COLOR_BGRA2BGR);

另一种解决方案是手动将所需数据手动复制到BGR矩阵中(4是BGRA中的通道数):

// Create the BGR (3 channel) matrix
cv::Mat3b frameBGR(height, width);


for(int r = 0; r < height; ++r) {
    for(int c = 0; c < width; ++c) {

        // Copy only BGR
        frameBGR(r,c)[0] = buffer((r * c * 4) + (c * 4) + 0); // Blue
        frameBGR(r,c)[1] = buffer((r * c * 4) + (c * 4) + 1); // Green
        frameBGR(r,c)[2] = buffer((r * c * 4) + (c * 4) + 2); // Red

        // Don't consider Alpha channel
        // buffer((r * c * 4) + (c * 4) + 3); // Alpha
    }
}