QT處理圖像用的是QImage類,而OpenCV處理圖像則是使用cv::Mat類,如果要在QT中運用OpenCV,那就需要進行QImage和Mat的相互轉換。
// cv::Mat轉換成QImage
QImage cvMat2QImage(const Mat& mat)
{
if (mat.type() == CV_8UC1) // 單通道
{
QImage image(mat.cols, mat.rows, QImage::Format_Indexed8);
image.setColorCount(256); // 灰度級數256
for (int i = 0; i < 256; i++)
{
image.setColor(i, qRgb(i, i, i));
}
uchar *pSrc = mat.data; // 複製mat數據
for (int row = 0; row < mat.rows; row++)
{
uchar *pDest = image.scanLine(row);
memcpy(pDest, pSrc, mat.cols);
pSrc += mat.step;
}
return image;
}
else if (mat.type() == CV_8UC3) // 3通道
{
const uchar *pSrc = (const uchar*)mat.data; // 複製像素
QImage image(pSrc, mat.cols, mat.rows, mat.step, QImage::Format_RGB888); // R, G, B 對應 0,1,2
return image.rgbSwapped(); // rgbSwapped是爲了顯示效果色彩好一些。
}
else if (mat.type() == CV_8UC4) // 4通道
{
const uchar *pSrc = (const uchar*)mat.data; // 複製像素
QImage image(pSrc, mat.cols, mat.rows, mat.step, QImage::Format_ARGB32); // B,G,R,A 對應 0,1,2,3
return image.copy();
}
else
{
return QImage();
}
}
// QImage轉換成cv::Mat
Mat QImage2cvMat(QImage image)
{
Mat mat;
switch (image.format())
{
case QImage::Format_ARGB32:
case QImage::Format_RGB32:
case QImage::Format_ARGB32_Premultiplied:
mat = Mat(image.height(), image.width(), CV_8UC4, (void*)image.constBits(), image.bytesPerLine());
break;
case QImage::Format_RGB888:
mat = Mat(image.height(), image.width(), CV_8UC3, (void*)image.constBits(), image.bytesPerLine());
cv::cvtColor(mat, mat, CV_BGR2RGB);
break;
case QImage::Format_Indexed8:
mat = Mat(image.height(), image.width(), CV_8UC1, (void*)image.constBits(), image.bytesPerLine());
break;
}
return mat;
}