yolov8_utils.h

Go to the documentation of this file.

#pragma once
#include<iostream>
#include <numeric>
#include<opencv2/opencv.hpp>
 
#define ORT_OLD_VISON 13  
 
namespace conscience_core::detector_engine::yolov8 {
struct OutputSeg {
    int id;            
    float confidence;  
    cv::Rect box;       
    cv::Mat boxMask;       
};
struct MaskParams {
    int netWidth = 640;
    int netHeight = 640;
    float maskThreshold = 0.5;
    cv::Size srcImgShape;
    cv::Vec4d params;
 
};
bool CheckModelPath(std::string modelPath);
bool CheckParams(int netHeight, int netWidth, const int* netStride, int strideSize); 
void DrawPred(cv::Mat& img,
    std::vector<OutputSeg> result,
    std::vector<std::string> classNames,
    std::vector<cv::Scalar> color,
    bool isVideo = false
);
void LetterBox(const cv::Mat& image, cv::Mat& outImage,
    cv::Vec4d& params, //[ratio_x,ratio_y,dw,dh]
    const cv::Size& newShape = cv::Size(640, 640),
    bool autoShape = false,
    bool scaleFill = false,
    bool scaleUp = true,
    int stride = 32,
    const cv::Scalar& color = cv::Scalar(114, 114, 114));
void GetMask(const cv::Mat& maskProposals, const cv::Mat& maskProtos, std::vector<OutputSeg>& output, const MaskParams& maskParams);
void GetMask2(const cv::Mat& maskProposals, const cv::Mat& maskProtos, OutputSeg& output, const MaskParams& maskParams);
 
 
 
}