39 lines
913 B
C
39 lines
913 B
C
#ifndef __XMYOLOV5N_H__
|
|
#define __XMYOLOV5N_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 "yolov5.h"
|
|
#include "svp_comm.h"
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
typedef struct {
|
|
yolov5_detect_param *detect_info;
|
|
xmedia_void *priv;
|
|
} svp_yolov5n_context;
|
|
|
|
xmedia_s32 xmyolov5n_detect_init(xmedia_void *context, const xmedia_svp_task_cfg cfg);
|
|
|
|
xmedia_s32 xmyolov5n_detect_uninit(xmedia_void *context);
|
|
|
|
xmedia_s32 xmyolov5n_detect_set_attr(xmedia_void *context, const xmedia_svp_yolov5_attr *task_attr);
|
|
|
|
xmedia_s32 xmyolov5n_detect_get_attr(xmedia_void *context, xmedia_svp_yolov5_attr *task_attr);
|
|
|
|
xmedia_s32 xmyolov5n_detect_process(xmedia_void *context, const xmedia_svp_task_input *input, xmedia_svp_yolov5_output *output);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif
|