Search before asking

[X] I have searched the YOLOv5 issues and found no similar bug report.

YOLOv5 Component

No response

Bug

Tried onnx version 17 and version 12 still the same error Screenshot from 2024-08-19 00-08-52

Here is the code

include

include <opencv2/opencv.hpp>

std::vector load_class_list() { std::vector class_list; std::ifstream ifs("/home/suman-singh/Downloads/yolov5-master/temp.txt"); std::string line; while (getline(ifs, line)) { class_list.push_back(line); } return class_list; }

void load_net(cv::dnn::Net &net, bool is_cuda) { auto result = cv::dnn::readNet("/home/suman-singh/Downloads/yolov5-master/yolov5s.onnx"); if (is_cuda) { std::cout << "Attempty to use CUDA\n"; result.setPreferableBackend(cv::dnn::DNN_BACKEND_CUDA); result.setPreferableTarget(cv::dnn::DNN_TARGET_CUDA_FP16); } else { std::cout << "Running on CPU\n"; result.setPreferableBackend(cv::dnn::DNN_BACKEND_OPENCV); result.setPreferableTarget(cv::dnn::DNN_TARGET_CPU); } net = result; }

const std::vector colors = {cv::Scalar(255, 255, 0), cv::Scalar(0, 255, 0), cv::Scalar(0, 255, 255), cv::Scalar(255, 0, 0)};

const float INPUT_WIDTH = 640.0; const float INPUT_HEIGHT = 640.0; const float SCORE_THRESHOLD = 0.2; const float NMS_THRESHOLD = 0.4; const float CONFIDENCE_THRESHOLD = 0.4;

struct Detection { int class_id; float confidence; cv::Rect box; };

cv::Mat format_yolov5(const cv::Mat &source) { int col = source.cols; int row = source.rows; int _max = MAX(col, row); cv::Mat result = cv::Mat::zeros(_max, _max, CV_8UC3); source.copyTo(result(cv::Rect(0, 0, col, row))); return result; }

void detect(cv::Mat &image, cv::dnn::Net &net, std::vector &output, const std::vector &className) { cv::Mat blob;

auto input_image = format_yolov5(image);

cv::dnn::blobFromImage(input_image, blob, 1./255., cv::Size(INPUT_WIDTH, INPUT_HEIGHT), cv::Scalar(), true, false);
net.setInput(blob);
std::vector<cv::Mat> outputs;
net.forward(outputs, net.getUnconnectedOutLayersNames());

float x_factor = input_image.cols / INPUT_WIDTH;
float y_factor = input_image.rows / INPUT_HEIGHT;

float *data = (float *)outputs[0].data;

const int dimensions = 85;
const int rows = 25200;

std::vector<int> class_ids;
std::vector<float> confidences;
std::vector<cv::Rect> boxes;

for (int i = 0; i < rows; ++i) {

    float confidence = data[4];
    if (confidence >= CONFIDENCE_THRESHOLD) {

        float * classes_scores = data + 5;
        cv::Mat scores(1, className.size(), CV_32FC1, classes_scores);
        cv::Point class_id;
        double max_class_score;
        minMaxLoc(scores, 0, &max_class_score, 0, &class_id);
        if (max_class_score > SCORE_THRESHOLD) {

            confidences.push_back(confidence);

            class_ids.push_back(class_id.x);

            float x = data[0];
            float y = data[1];
            float w = data[2];
            float h = data[3];
            int left = int((x - 0.5 * w) * x_factor);
            int top = int((y - 0.5 * h) * y_factor);
            int width = int(w * x_factor);
            int height = int(h * y_factor);
            boxes.push_back(cv::Rect(left, top, width, height));
        }

    }

    data += 85;

}

std::vector<int> nms_result;
cv::dnn::NMSBoxes(boxes, confidences, SCORE_THRESHOLD, NMS_THRESHOLD, nms_result);
for (int i = 0; i < nms_result.size(); i++) {
    int idx = nms_result[i];
    Detection result;
    result.class_id = class_ids[idx];
    result.confidence = confidences[idx];
    result.box = boxes[idx];
    output.push_back(result);
}

}

int main(int argc, char **argv) {

std::vector<std::string> class_list = load_class_list();

cv::Mat frame;
cv::VideoCapture capture(0);

cv::dnn::Net net;
load_net(net, true);

while (true)
{
    capture.read(frame);

    std::vector<Detection> output;
    detect(frame, net, output, class_list);

    int detections = output.size();

    for (int i = 0; i < detections; ++i)
    {

        auto detection = output[i];
        auto box = detection.box;
        auto classId = detection.class_id;
        const auto color = colors[classId % colors.size()];
        cv::rectangle(frame, box, color, 3);

        cv::rectangle(frame, cv::Point(box.x, box.y - 20), cv::Point(box.x + box.width, box.y), color, cv::FILLED);
        cv::putText(frame, class_list[classId].c_str(), cv::Point(box.x, box.y - 5), cv::FONT_HERSHEY_SIMPLEX, 0.5, cv::Scalar(0, 0, 0));
    }

    cv::imshow("output", frame);

    if (cv::waitKey(1) != -1)
    {
        capture.release();
        std::cout << "finished by user\n";
        break;
    }
}

return 0;

}

Environment

No response

Minimal Reproducible Example

i want my webcam to open and take video but it closes immediately

Additional

Plzz help fast

Are you willing to submit a PR?

[X] Yes I'd like to help by submitting a PR!

ultralytics / yolov5

A Error which blast my mind.... #13265