120 lines
3.3 KiB
C++
120 lines
3.3 KiB
C++
#pragma once
|
||
|
||
//
|
||
#ifdef YOLO_EXPORTS
|
||
#define YOLO_API __declspec(dllexport)
|
||
#else
|
||
#define YOLO_API __declspec(dllimport)
|
||
#endif
|
||
|
||
#include <vector>
|
||
|
||
//
|
||
struct Detection
|
||
{
|
||
int class_id;
|
||
float score;
|
||
int x;
|
||
int y;
|
||
int width;
|
||
int height;
|
||
};
|
||
|
||
extern "C" {
|
||
|
||
// ========================================================================
|
||
//
|
||
// ========================================================================
|
||
|
||
/**
|
||
* @brief [新增] 创建一个检测器实例 (有状态)
|
||
* @details 此函数加载ONNX模型并初始化Session，返回一个句柄。
|
||
* @param model_path ONNX模型的绝对路径
|
||
* @param input_width 模型输入宽度 (例如 640)
|
||
* @param input_height 模型输入高度 (例如 640)
|
||
* @return void* 指向检测器实例的句柄。如果失败则返回 nullptr。
|
||
*/
|
||
YOLO_API void* create_detector(
|
||
const wchar_t* model_path,
|
||
int input_width,
|
||
int input_height
|
||
);
|
||
|
||
/**
|
||
* @brief [新增] 释放由 create_detector 创建的检测器实例
|
||
* @param detector_handle 要释放的句柄
|
||
*/
|
||
YOLO_API void free_detector(void* detector_handle);
|
||
|
||
/**
|
||
* @brief [新增] 使用已加载的检测器执行检测 (线程安全)
|
||
* @details
|
||
* @param detector_handle 由 create_detector 返回的句柄
|
||
* @param image_bytes 指向图像数据(BGR格式)的指针
|
||
* @param image_width 图像宽度
|
||
* @param image_height 图像高度
|
||
* @param out_detections [输出] Detection结构体数组指针
|
||
* @param out_detections_count [输出] 检测到的物体数量
|
||
* @param conf_threshold 置信度阈值
|
||
* @param iou_threshold NMS的IOU阈值
|
||
* @return int 0表示成功
|
||
*/
|
||
YOLO_API int perform_detection_on_session(
|
||
void* detector_handle,
|
||
unsigned char* image_bytes,
|
||
int image_width,
|
||
int image_height,
|
||
Detection** out_detections,
|
||
int* out_detections_count,
|
||
float conf_threshold,
|
||
float iou_threshold
|
||
);
|
||
|
||
|
||
// ========================================================================
|
||
//
|
||
// ========================================================================
|
||
|
||
/**
|
||
* @brief [保留] 原始的无状态检测函数 (非线程安全)
|
||
*/
|
||
YOLO_API int perform_detection(
|
||
const wchar_t* model_path,
|
||
unsigned char* image_bytes,
|
||
int image_width,
|
||
int image_height,
|
||
Detection** out_detections,
|
||
int* out_detections_count,
|
||
const char** class_names,
|
||
int class_names_count,
|
||
float conf_threshold, // 置信度阈值 (例如 0.5f)
|
||
float iou_threshold, // NMS的IOU阈值 (例如 0.5f)
|
||
int input_width, // 模型输入宽度 (例如 640)
|
||
int input_height // 模型输入高度 (例如 640)
|
||
);
|
||
|
||
/**
|
||
* @brief [保留] 释放由 perform_detection* 分配的内存
|
||
*/
|
||
YOLO_API void free_memory(Detection* detections);
|
||
|
||
/**
|
||
* @brief [保留] 绘制函数
|
||
*/
|
||
YOLO_API void draw_and_encode_image(
|
||
unsigned char* image_bytes,
|
||
int image_width,
|
||
int image_height,
|
||
const Detection* detections,
|
||
int detections_count,
|
||
const char** class_names,
|
||
int class_names_count,
|
||
unsigned char** out_image_bytes,
|
||
int* out_image_size
|
||
);
|
||
|
||
/**
|
||
* @brief [保留] 释放由 draw_and_encode_image 分配的内存
|
||
*/
|
||
YOLO_API void free_image_memory(unsigned char* image_bytes);
|
||
} |