递归火山软件开发平台

标题: YOLOv5检测器预测图像 [打印本页]

作者: zhh044 时间: 2024-2-25 17:36
标题: YOLOv5检测器预测图像
YOLOv5检测器预测图像如何调用gpu?我一做预测CPU就一直飙。

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

作者: zhh044 时间: 2024-2-26 01:44

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

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

作者: 赵曰天 时间: 2024-2-26 22:41

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

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

作者: zhh044 时间: 2024-2-27 00:49

赵曰天发表于 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
用yolo4啊

欢迎光临递归火山软件开发平台 (https://bbs.voldp.com/)