yolov8.h

Go to the documentation of this file.

#pragma once
#include<iostream>
#include<opencv2/opencv.hpp>
#include "yolov8_utils.h"
using std::string,std::vector;
namespace conscience_core::detector_engine::yolov8 {
class Yolov8 {
public:
    Yolov8() {
    }
    ~Yolov8() {}
 
    bool ReadModel(cv::dnn::Net& net, std::string& netPath, bool isCuda);
    bool Detect(const cv::Mat& srcImg, cv::dnn::Net& net, std::vector<OutputSeg>& output);
 
    int _netWidth = 640;  
    int _netHeight = 640; 
 
 
    std::vector<std::string> _className = { "person", "bicycle", "car", "motorcycle", "airplane", "bus", "train", "truck", "boat", "traffic light",
        "fire hydrant", "stop sign", "parking meter", "bench", "bird", "cat", "dog", "horse", "sheep", "cow",
        "elephant", "bear", "zebra", "giraffe", "backpack", "umbrella", "handbag", "tie", "suitcase", "frisbee",
        "skis", "snowboard", "sports ball", "kite", "baseball bat", "baseball glove", "skateboard", "surfboard",
        "tennis racket", "bottle", "wine glass", "cup", "fork", "knife", "spoon", "bowl", "banana", "apple",
        "sandwich", "orange", "broccoli", "carrot", "hot dog", "pizza", "donut", "cake", "chair", "couch",
        "potted plant", "bed", "dining table", "toilet", "tv", "laptop", "mouse", "remote", "keyboard", "cell phone",
        "microwave", "oven", "toaster", "sink", "refrigerator", "book", "clock", "vase", "scissors", "teddy bear",
        "hair drier", "toothbrush" };
private:
    float _classThreshold = 0.50;
    float _nmsThreshold = 0.32;
};
}