opencv学习（三）

2022-12-16

cv::Mat与QImage相互转化

/****************************************************************************
** file: convert.h
** brief: cv::Mat 与 QImage 之间的转换函数。在 Qt 程序中使用 OpenCV 时有用。
** Copyright (C) LiYuan
** Author: LiYuan
** E-Mail: 18069211#qq(.)com
** Version 2.0.1
** Last modified: 2019.01.11
** Modified By: LiYuan
****************************************************************************/


#ifndef QT_OPENCV_CONVERT_H
#define QT_OPENCV_CONVERT_H

#include <QtCore/QDebug>
#include <QtGui/QImage>
#include "opencv2/core/core.hpp"
#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/opencv.hpp"
/**
 * @brief 将 OpenCV 的 cv::Mat 类型图像转换为 QImage 类型
 * @param mat 待转换的图像，支持 CV_8UC1、CV_8UC3、CV_8UC4 三种OpenCV 的数据类型
 * @param clone true 表示与 Mat 不共享内存，更改生成的 mat 不会影响原始图像，false 则会与 mat 共享内存
 * @param rb_swap 只针对 CV_8UC3 格式，如果 true 则会调换 R 与 B RGB->BGR，如果共享内存的话原始图像也会发生变化
 * @return 转换后的 QImage 图像
*/
QImage cvMat2QImage(const cv::Mat& mat, bool clone = true, bool rb_swap = true);

/**
 * @brief 将 QImage 的类型图像转换为 cv::Mat 类型
 * @param image 待转换的图像，支持 Format_Indexed8/Format_Grayscale、24 位彩色、32 位彩色格式，
 * @param clone true 表示与 QImage 不共享内存，更改生成的 mat 不会影响原始图像，false 则会与 QImage 共享内存
 * @param rg_swap 只针对 RGB888 格式，如果 true 则会调换 R 与 B RGB->BGR，如果共享内存的话原始图像也会发生变化
 * @return 转换后的 cv::Mat 图像
*/
cv::Mat QImage2cvMat(QImage &image, bool clone = false, bool rb_swap = true);

#endif

/****************************************************************************
** file: convert.cpp
** brief: cv::Mat 与 QImage 之间的转换函数。在 Qt 程序中使用 OpenCV 时有用。
** Copyright (C) LiYuan
** Author: LiYuan
** E-Mail: 18069211#qq(.)com
** Version 2.0.1
** Last modified: 2019.01.11
** Modified By: LiYuan
****************************************************************************/

#include "convert.h"

QImage cvMat2QImage(const cv::Mat& mat, bool clone, bool rb_swap)
{
    const uchar *pSrc = (const uchar*)mat.data;
    // 8-bits unsigned, NO. OF CHANNELS = 1
    if(mat.type() == CV_8UC1)
    {
        //QImage image(mat.cols, mat.rows, QImage::Format_Grayscale8);
        QImage image(pSrc, mat.cols, mat.rows, mat.step, QImage::Format_Grayscale8);
        if(clone) return image.copy();
        return image;
    }
    // 8-bits unsigned, NO. OF CHANNELS = 3
    else if(mat.type() == CV_8UC3)
    {
        // Create QImage with same dimensions as input Mat
        QImage image(pSrc, mat.cols, mat.rows, mat.step, QImage::Format_RGB888);
        if(clone)
        {
            if(rb_swap) return image.rgbSwapped();
            return image.copy();
        }
        else
        {
            if(rb_swap)
            {
                cv::cvtColor(mat, mat, CV_BGR2RGB);
            }
            return image;
        }

    }
    else if(mat.type() == CV_8UC4)
    {
        qDebug() << "CV_8UC4";
        QImage image(pSrc, mat.cols, mat.rows, mat.step, QImage::Format_ARGB32);
        if(clone) return image.copy();
        return image;
    }
    else
    {
        qDebug() << "ERROR: Mat could not be converted to QImage.";
        return QImage();
    }
}

cv::Mat QImage2cvMat(QImage &image, bool clone, bool rb_swap)
{
    cv::Mat mat;
    //qDebug() << image.format();
    switch(image.format())
    {
    case QImage::Format_ARGB32:
    case QImage::Format_RGB32:
    case QImage::Format_ARGB32_Premultiplied:
        mat = cv::Mat(image.height(), image.width(), CV_8UC4, (void *)image.constBits(), image.bytesPerLine());
        if(clone)  mat = mat.clone();
        break;
    case QImage::Format_RGB888:
        mat = cv::Mat(image.height(), image.width(), CV_8UC3, (void *)image.constBits(), image.bytesPerLine());
        if(clone)  mat = mat.clone();
        if(rb_swap) cv::cvtColor(mat, mat, CV_BGR2RGB);
        break;
    case QImage::Format_Indexed8:
    case QImage::Format_Grayscale8:
        mat = cv::Mat(image.height(), image.width(), CV_8UC1, (void *)image.bits(), image.bytesPerLine());
        if(clone)  mat = mat.clone();
        break;
    }
    return mat;
}

使用

void test1()
{
    cv::Mat mat = cv::imread("Q:\\Koala.jpg", cv::IMREAD_UNCHANGED);
    cv::cvtColor(mat, mat, CV_BGR2BGRA);
    QImage image = cvMat2QImage(mat);
    qDebug() << (mat.type() == CV_8UC4);
    cvNamedWindow("cvMat2QImage RGB32", CV_WINDOW_AUTOSIZE);
    imshow("cvMat2QImage RGB32", mat);
    QLabel label;
    label.setPixmap(QPixmap::fromImage(image));
    label.show();
    cv::waitKey(10000);
}
void test2()
{
    cv::Mat mat = cv::imread("Q:\\Koala.jpg", cv::IMREAD_UNCHANGED);
    cv::cvtColor(mat, mat, CV_BGR2GRAY);
    QImage image = cvMat2QImage(mat);
    cvNamedWindow("cvMat2QImage gray", CV_WINDOW_AUTOSIZE);
    imshow("cvMat2QImage gray", mat);
    QLabel label;
    label.setPixmap(QPixmap::fromImage(image));
    label.show();
    cv::waitKey(10000);
}
void test3()
{
    QImage image("Q:\\Koala.jpg");
    image = image.convertToFormat(QImage::Format_RGB32);
    cv::Mat mat = QImage2cvMat(image);
    //cv::cvtColor(mat, mat, CV_BGR2RGB);
    imshow("QImage2cvMat RGB32", mat);
    cv::waitKey(10000);
}
 
void test4()
{
    QImage image("Q:\\Koala.jpg");
    image = image.convertToFormat(QImage::Format_RGB888);
    cv::Mat mat = QImage2cvMat(image);
    imshow("QImage2cvMat RGB24", mat);
    cv::waitKey(10000);
}
void test5()
{
    QImage image("Q:\\Koala.jpg");
    image = image.convertToFormat(QImage::Format_Indexed8);
    cv::Mat mat = QImage2cvMat(image);
    imshow("QImage2cvMat Indexed8", mat);
    cv::waitKey(10000);
}
int main(int argc, char *argv[])
{
    QApplication a(argc, argv);
    //test1();
    //test2();
    //test3();
    //test4();
    //test5();
    test1();
    return a.exec();
}

转载

1	https://blog.csdn.net/liyuanbhu/article/details/86307283