Renamed opi_player as rtsp_ai_player

2026-04-22 10:56:34 +00:00 · 2024-07-14 19:52:45 +02:00
parent 813251b170
commit 8a7d681e5b
40 changed files with 43 additions and 0 deletions
@@ -0,0 +1,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);
+}