QT+opencv实现截图并跟踪图像

2023-12-31 14:12:28

版本：

　　QT 5.9

　　OPENCV： 4.5

实现功能：打开摄像头，并截取通过鼠标拖动截取视频的一部分图片，然后保存，通过模板匹配的方式在视频中跟踪图片。

使用到技术：QT显示视频，QT打开文件，QT实现鼠标拖动截图，QT实现图片保存，QT实现图片模板匹配

第一部分：

　　打开视频，并通过两个label进行显示。通过定时器周期读取数据

    //打开摄像头

      videoCap = new VideoCapture(ui->comboBox->currentIndex());

      this->videoFlag = true;      timer->start(33); //33ms

*******************************************************
************************************************
//显示图片

 videoCap->read(gInMat);

    //qDebug()<<matframe.rows<<endl;

    //qDebug()<<matframe.cols<<endl;

    gInMat.copyTo(gOutMat);

    emit setImageCmd(1,NULL); //发送信号给线程处理数据

    QImage image = cvMat2Image(gInMat);

    if(displayFlag)

            displayFlag = false;

            disImage = image; //赋值给ima

            MatImage = gInMat;//保存一个mat,用于截图使用

            DisplayFrom->show();

    ui->label->setPixmap(QPixmap::fromImage(image));

//图像转换借鉴

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,COLOR_BGR2RGB);

            break;

        case QImage::Format_Indexed8:

            mat = Mat(image.height(),image.width(),CV_8UC1,(void *)image.constBits(),image.bytesPerLine());

            break;

        default:

            break;

    return mat;

QImage cvMat2Image(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)

        uchar *pSrc = mat.data;

        QImage image(pSrc,mat.cols,mat.rows,mat.step,QImage::Format_RGB888);

        return image.rgbSwapped();

    else if(mat.type() == CV_8UC4)

        const uchar *pSrc = (const uchar *)mat.data;

        QImage image(pSrc,mat.cols,mat.rows,mat.step,QImage::Format_ARGB32);

        //return image.copy();

        return image;

    else

        QImage image;

        return image;

}
/*****************************************/
鼠标截图

void DisplayDialog::on_dispushButton_clicked()

    if(isrectFlag == false)

        QMessageBox warn;

        qDebug()<<"please2";

        warn.warning(this,"wanging!","please 2");

    else

        if(saveflag)

            int x,y,w,h;

            x = lastPoint.x();

            y = lastPoint.y();

            w = endPoint.x() - x;

            h = endPoint.y() - y;

            Rect rect(x,y,w,h);

            Mat ROI = MatImage(rect);

             qDebug()<<"write,"<<w<<","<<h<<endl;

           imwrite(savefilename.toStdString(),ROI);

        else

            QMessageBox warn;

            qDebug()<<"please1";

            warn.warning(this,"wanging!","please 1");

   // isdRrawing = true;

void DisplayDialog::mousePressEvent(QMouseEvent *event)

    if(event->button() == Qt::LeftButton)

        isdRrawing = true;

        lastPoint = event->pos();

void DisplayDialog::mouseMoveEvent(QMouseEvent *event)

    if(event->buttons() == Qt::LeftButton)

          endPoint = event->pos();

          update();

void DisplayDialog::mouseReleaseEvent(QMouseEvent *event)

   // qDebug()<<"release111"<<endl;

    if(event->button() == Qt::LeftButton)

        endPoint = event->pos();

      //  qDebug()<<"release"<<endl;

        isrectFlag = true;

        isdRrawing = false;

      //  update();

}
//模板匹配

 result_cols = gOutMat.cols - temp1.cols +1;

    result_rows = gOutMat.rows - temp1.rows+1;

    result.create(result_rows,result_cols,CV_32FC1);

  //  bool Method_accepts_mask = (CV_)

    matchTemplate(gOutMat,temp1,result,Method);//进行匹配

    normalize(result,result,0,1,NORM_MINMAX,-1,Mat());//进行标准化

    minMaxLoc(result,&minVal,&maxVal,&minLoc,&maxLoc,Mat());

    if(Method == TM_SQDIFF ||Method == TM_SQDIFF_NORMED)

            matchLoc =minLoc;

    else

        matchLoc = maxLoc;

    rectangle(gOutMat,matchLoc,Point(matchLoc.x+temp1.cols,matchLoc.y+temp1.rows),Scalar(0,0,255),2,8,0);

  //  rectangle(result,matchLoc,POint(matchLoc.x+temp1.cols,matchLoc.y+temp1.rows),Scalar::all(0),2,8,0);

码农公寓