在QT中做项目时如果需要对图像进行处理,经常需要用到Mat类型与QImage类型的项目转换。我最近在利用QT做的视频处理,应用的是VideoCapture类,最后在现实环节需要将Mat转换成QImage类型。视频流态的处理,对单帧的响应速度要求较高,一开始我也是直接用的别人的函数,最后发现Mat转QImage时间太长,做了一个小改进。2592*2048*3的图像的转换速度从平均50ms变成10ms(i5-9300H)。原因主要OpenCV的BGR类型与QImage的RGB类型不同造成,因此在转换时需要将BGR转换为RGB类型。
我的主要改变是在使用QImage::rgbSwapped()函数前使用:cv::cvtColor(mat,mat,cv::COLOR_BGR2RGB),可以有效提高效率。
下面直接放代码:
QImage cvMat2QImage(const cv::Mat &mat)
{
// 8-bits unsigned, NO. OF CHANNELS = 1
if(mat.type() == CV_8UC1)
{
QImage image(mat.cols, mat.rows, QImage::Format_Indexed8);
image.setColorCount(256);
for(int i = 0; i < 256; i++)
{
image.setColor(i, qRgb(i, i, i));
}
uchar *pSrc = mat.data;
for(int row = 0; row < mat.rows; row ++)
{
uchar *pDest = image.scanLine(row);
memcpy(pDest, pSrc, mat.cols);
pSrc += mat.step;
}
return image;
}
// 8-bits unsigned, NO. OF CHANNELS = 3
else if(mat.type() == CV_8UC3)
{
// Copy input Mat
cv::cvtColor(mat,mat,cv::COLOR_BGR2RGB);
const uchar *pSrc = (const uchar*)mat.data;
QImage image(pSrc, mat.cols, mat.rows, mat.step, QImage::Format_RGB888);
//image = image.rgbSwapped();
return image.copy();
}
else if(mat.type() == CV_8UC4)
{
// Copy input Mat
const uchar *pSrc = (const uchar*)mat.data;
QImage image(pSrc, mat.cols, mat.rows, mat.step, QImage::Format_ARGB32);
return image.copy();
}
else
{
qDebug() << "ERROR: Mat could not be converted to QImage.";
return QImage();
}
}