Qt之dnn_yolov4识别物体

98 篇文章 5 订阅

在这里插入图片描述
代码如下:

// CUDA support
if (1)
{
    net.setPreferableTarget(cv::dnn::DNN_TARGET_CUDA);
    net.setPreferableBackend(cv::dnn::DNN_BACKEND_CUDA);
}
// [4]载入理图像
cv::Mat frame;
cv::Mat inputBlob;
double timer = 0;
if (1)
{
    // frame = QImage2Mat(m_image); //图像
    frame = cv::imread(m_imagestr.toStdString().c_str());
    if(frame.empty()){
        qDebug()<<"图片打开失败";
        return;
    }
    qDebug()<<"sendImage1";
    inputBlob = cv::dnn::blobFromImage(frame, 1.0 / 255, cv::Size(608,608), cv::Scalar());
    net.setInput(inputBlob);
    qDebug()<<"sendImage2";
    // 启动计时器
    timer = (double)cv::getTickCount();
    qDebug()<<"sendImage3";
    // [5] 模型推理 获取未连接输出层
    std::vector<cv::String> outNames = net.getUnconnectedOutLayersNames();
    qDebug()<<"sendImage4";
    std::vector<cv::Mat> outs;
    qDebug()<<"sendImage5"<<outNames.size();
    if(outNames.size()!=3){
        qDebug()<<"图片通道数不是3的倍数";
        return;
    }
    // 前向传播
    net.forward(outs, outNames);
    qDebug()<<"sendImage5";
    // 计算耗时:
    t = ((double)cv::getTickCount() - timer) / cv::getTickFrequency();  //获得时间,单位是秒;
    string label = cv::format("YOLO v4");
    cv::putText(frame, label, cv::Point(15, 15), cv::FONT_HERSHEY_COMPLEX, 0.5, cv::Scalar(0, 0, 255));
    qDebug()<<"sendImage3";
    // [6]结果后处理
    vector<cv::Rect> boxes;
    vector<int> classIds;
    vector<int> indices;
       qDebug()<<"postProcess1";
    postProcess(net, frame, outs, boxes, classIds, indices, confidenceThreshold, nmsThreshold);
      qDebug()<<"postProcess2";
    // [7]绘制显示结果
    drawPred(frame, boxes, classIds, indices, classNamesVec);
    QImage t_image = cvMat2QImage(frame);
    emit sendImage(t_image);
    qDebug()<<"sendImage"<<t_image.width();
}

在这里插入图片描述

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

qt5_qt6

你的鼓励是我们创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值