Qt結(jié)合OpenCV部署yolov5的實現(xiàn)

第Qt結(jié)合OpenCV部署yolov5的實現(xiàn)connect(timer,SIGNAL(timeout()),this,SLOT(readFrame()));

ui-startdetect-setEnabled(false);

ui-stopdetect-setEnabled(false);

Init();

MainWindow::~MainWindow()

capture-release();

deletecapture;

delete[]yolo_nets;

deleteyolov5;

deleteui;

voidMainWindow::Init()

capture=newcv::VideoCapture();

yolo_nets=newNetConfig[4]{

{0.5,0.5,0.5,"yolov5s"},

{0.6,0.6,0.6,"yolov5m"},

{0.65,0.65,0.65,"yolov5l"},

{0.75,0.75,0.75,"yolov5x"}

conf=yolo_nets[0];

yolov5=newYOLOV5();

yolov5-Initialization(conf);

ui-textEditlog-append(QStringLiteral("默認(rèn)模型類別：yolov5sargs:%1%2%3")

.arg(conf.nmsThreshold)

.arg(conf.objThreshold)

.arg(conf.confThreshold));

voidMainWindow::readFrame()

cv::Matframe;

capture-read(frame);

if(frame.empty())return;

autostart=std::chrono::steady_clock::now();

yolov5-detect(frame);

autoend=std::chrono::steady_clock::now();

std::chrono::durationdouble,std::millielapsed=end-start;

ui-textEditlog-append(QString("cost_time:%1ms").arg(elapsed.count()));

//doublet0=static_castdouble(cv::getTickCount());

//yolov5-detect(frame);

//doublet1=static_castdouble(cv::getTickCount());

//ui-textEditlog-append(QStringLiteral("cost_time:%1").arg((t1-t0)/cv::getTickFrequency()));

cv::cvtColor(frame,frame,cv::COLOR_BGR2RGB);

QImagerawImage=QImage((uchar*)(frame.data),frame.cols,frame.rows,frame.step,QImage::Format_RGB888);

ui-label-setPixmap(QPixmap::fromImage(rawImage));

voidMainWindow::on_openfile_clicked()

QStringfilename=QFileDialog::getOpenFileName(this,QStringLiteral("打開文件"),".","*.mp4*.avi;;*.png*.jpg*.jpeg*.bmp");

if(!QFile::exists(filename)){

return;

ui-statusbar-showMessage(filename);

QMimeDatabasedb;

QMimeTypemime=db.mimeTypeForFile(filename);

if(().startsWith("image/")){

cv::Matsrc=cv::imread(filename.toLatin1().data());

if(src.empty()){

ui-statusbar-showMessage("圖像不存在！");

return;

cv::Mattemp;

if(src.channels()==4)

cv::cvtColor(src,temp,cv::COLOR_BGRA2RGB);

elseif(src.channels()==3)

cv::cvtColor(src,temp,cv::COLOR_BGR2RGB);

else

cv::cvtColor(src,temp,cv::COLOR_GRAY2RGB);

autostart=std::chrono::steady_clock::now();

yolov5-detect(temp);

autoend=std::chrono::steady_clock::now();

std::chrono::durationdouble,std::millielapsed=end-start;

ui-textEditlog-append(QString("cost_time:%1ms").arg(elapsed.count()));

QImageimg=QImage((uchar*)(temp.data),temp.cols,temp.rows,temp.step,QImage::Format_RGB888);

ui-label-setPixmap(QPixmap::fromImage(img));

ui-label-resize(ui-label-pixmap()-size());

filename.clear();

}elseif(().startsWith("video/")){

capture-open(filename.toLatin1().data());

if(!capture-isOpened()){

ui-textEditlog-append("failtoopenMP4!");

return;

IsDetect_ok+=1;

if(IsDetect_ok==2)

ui-startdetect-setEnabled(true);

ui-textEditlog-append(QString::fromUtf8("Openvideo:%1succesfully!").arg(filename));

//獲取整個幀數(shù)QStringLiteral

longtotalFrame=capture-get(cv::CAP_PROP_FRAME_COUNT);

ui-textEditlog-append(QStringLiteral("整個視頻共%1幀").arg(totalFrame));

ui-label-resize(QSize(capture-get(cv::CAP_PROP_FRAME_WIDTH),capture-get(cv::CAP_PROP_FRAME_HEIGHT)));

//設(shè)置開始幀()

longframeToStart=0;

capture-set(cv::CAP_PROP_POS_FRAMES,frameToStart);

ui-textEditlog-append(QStringLiteral("從第%1幀開始讀").arg(frameToStart));

//獲取幀率

doublerate=capture-get(cv::CAP_PROP_FPS);

ui-textEditlog-append(QStringLiteral("幀率為:%1").arg(rate));

voidMainWindow::on_loadfile_clicked()

QStringonnxFile=QFileDialog::getOpenFileName(this,QStringLiteral("選擇模型"),".","*.onnx");

if(!QFile::exists(onnxFile)){

return;

ui-statusbar-showMessage(onnxFile);

if(!yolov5-loadModel(onnxFile.toLatin1().data())){

ui-textEditlog-append(QStringLiteral("加載模型失敗！"));

return;

IsDetect_ok+=1;

ui-textEditlog-append(QString::fromUtf8("OpenonnxFile:%1succesfully!").arg(onnxFile));

if(IsDetect_ok==2)

ui-startdetect-setEnabled(true);

voidMainWindow::on_startdetect_clicked()

timer-start();

ui-startdetect-setEnabled(false);

ui-stopdetect-setEnabled(true);

ui-openfile-setEnabled(false);

ui-loadfile-setEnabled(false);

ui-comboBox-setEnabled(false);

ui-textEditlog-append(QStringLiteral("================\n"

"開始檢測\n"

"================\n"));

voidMainWindow::on_stopdetect_clicked()

ui-startdetect-setEnabled(true);

ui-stopdetect-setEnabled(false);

ui-openfile-setEnabled(true);

ui-loadfile-setEnabled(true);

ui-comboBox-setEnabled(true);

timer-stop();

ui-textEditlog-append(QStringLiteral("================\n"

"停止檢測\n"

"================\n"));

voidMainWindow::on_comboBox_activated(constQStringarg1)

if(arg1.contains("s")){

conf=yolo_nets[0];

}elseif(arg1.contains("m")){

conf=yolo_nets[1];

}elseif(arg1.contains("l")){

conf=yolo_nets[2];

}elseif(arg1.contains("x")){

conf=yolo_nets[3];}

yolov5-Initialization(conf);

ui-textEditlog-append(QStringLiteral("使用模型類別：%1args:%2%3%4")

.arg(arg1)

.arg(con