utils.h 1.12 KB
Newer Older
change's avatar
change committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
#pragma once
#include <codecvt>
#include <fstream>
#include <opencv2/opencv.hpp>

struct Yolov8Result
{
    cv::Rect box;
    cv::Mat boxMask; // mask in box
    float conf{};
    int classId{};
};

namespace utils
{
    static std::vector<cv::Scalar> colors;

    size_t vectorProduct(const std::vector<int64_t> &vector);
    std::wstring charToWstring(const char *str);
    std::vector<std::string> loadNames(const std::string &path);
    void visualizeDetection(cv::Mat &image, std::vector<Yolov8Result> &results,
                            const std::vector<std::string> &classNames);

    void letterbox(const cv::Mat &image, cv::Mat &outImage,
                   const cv::Size &newShape,
                   const cv::Scalar &color,
                   bool auto_,
                   bool scaleFill,
                   bool scaleUp,
                   int stride);

    void scaleCoords(cv::Rect &coords, cv::Mat &mask,
                     const float maskThreshold,
                     const cv::Size &imageShape, const cv::Size &imageOriginalShape);

    template <typename T>
    T clip(const T &n, const T &lower, const T &upper);
}