zhh044 发表于 2024-2-25 17:36:42

YOLOv5检测器预测图像

YOLOv5检测器预测图像如何调用gpu?我一做预测CPU就一直飙。

赵曰天 发表于 2024-2-25 23:35:06

目前我也是没有办法调用，但是别人说在运行的机器上安装cuda，cudnn这些就可以了，但是我调用py的yolo是可以设置显卡推理的，火山的我就没有找到相应的代码或者命令。

zhh044 发表于 2024-2-26 01:44:45

赵曰天 发表于 2024-2-25 23:35
目前我也是没有办法调用，但是别人说在运行的机器上安装cuda，cudnn这些就可以了，但是我调用py的yolo是可 ...

我遇到的问题是，推理摄像头数据，可是运行一会儿就崩溃了。

赵曰天 发表于 2024-2-26 22:41:20

zhh044 发表于 2024-2-26 01:44
我遇到的问题是，推理摄像头数据，可是运行一会儿就崩溃了。

那这个我目前没遇到运行一会就崩溃的。

zhh044 发表于 2024-2-27 00:49:44

赵曰天 发表于 2024-2-26 22:41
那这个我目前没遇到运行一会就崩溃的。

#include <opencv2/opencv.hpp>
#include <iostream>

int main() {
    // 打开默认摄像头，设备索引为0
    cv::VideoCapture cap(0);

    if (!cap.isOpened()) {
      std::cerr << "无法打开摄像头" << std::endl;
      return -1;
    }

    // 创建一个窗口来显示摄像头数据
    cv::namedWindow("摄像头预览", cv::WINDOW_AUTOSIZE);

    while (true) {
      // 从摄像头捕获一帧图像
      cv::Mat frame;
      cap >> frame;

      if (frame.empty()) {
            std::cerr << "无法捕获图像" << std::endl;
            break;
      }

      // 在窗口中显示图像
      cv::imshow("摄像头预览", frame);

      // 等待用户按键，如果是'q'则退出循环
      char c = (char)cv::waitKey(25);
      if (c == 'q' || c == 27) {
            break;
      }
    }

    // 释放摄像头资源并关闭窗口
    cap.release();
    cv::destroyAllWindows();

    return 0;
}

zzr1122333 发表于 2024-2-28 15:00:28

用yolo4啊
页: [1]
查看完整版本: YOLOv5检测器预测图像