38 lines
883 B
C
38 lines
883 B
C
#ifndef __XMYOLOV8_H__
|
|
#define __XMYOLOV8_H__
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
|
|
#include "xmedia_svp_std.h"
|
|
#include "xmedia_svp.h"
|
|
#include "svp_nms.h"
|
|
#include "xmedia_svp_quantize.h"
|
|
#include "svp_core.h"
|
|
#include "yolov8.h"
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
typedef struct {
|
|
yolov8_detect_param *detect_info;
|
|
xmedia_void *priv;
|
|
} svp_yolov8_context;
|
|
|
|
xmedia_s32 xmyolov8_detect_init(xmedia_void *context, const xmedia_svp_task_cfg cfg);
|
|
|
|
xmedia_s32 xmyolov8_detect_uninit(xmedia_void *context);
|
|
|
|
xmedia_s32 xmyolov8_detect_set_attr(xmedia_void *context, const xmedia_svp_yolov8_attr *task_attr);
|
|
|
|
xmedia_s32 xmyolov8_detect_get_attr(xmedia_void *context, xmedia_svp_yolov8_attr *task_attr);
|
|
|
|
xmedia_s32 xmyolov8_detect_process(xmedia_void *context, const xmedia_svp_task_input *input, xmedia_svp_yolov8_output *output);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif
|