YOLOv5检测器预测图像

zhh044

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

赵曰天

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

zhh044

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

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

赵曰天

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

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

zhh044

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

用yolo4啊