mirror of
https://github.com/azaion/autopilot.git
synced 2026-04-22 21:46:33 +00:00
Tuomas Järvinen
45c19baa45
- autopilot -> drone_controller - rtsp_ai_player -> ai_controller - added top level qmake project file - updated documentation - moved small demo applications from tmp/ to misc/
39 lines
1.1 KiB
C++
39 lines
1.1 KiB
C++
#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);
|
|
}
|