FFmpeg
Functions
dnn_io_proc.c File Reference
#include "dnn_io_proc.h"
#include "libavutil/imgutils.h"
#include "libavutil/mem.h"
#include "libswscale/swscale.h"
#include "libavutil/avassert.h"
#include "libavutil/detection_bbox.h"

Go to the source code of this file.

Functions

static int get_datatype_size (DNNDataType dt)
 
int ff_proc_from_dnn_to_frame (AVFrame *frame, DNNData *output, void *log_ctx)
 
int ff_proc_from_frame_to_dnn (AVFrame *frame, DNNData *input, void *log_ctx)
 
static enum AVPixelFormat get_pixel_format (DNNData *data)
 
int ff_frame_to_dnn_classify (AVFrame *frame, DNNData *input, uint32_t bbox_index, void *log_ctx)
 
int ff_frame_to_dnn_detect (AVFrame *frame, DNNData *input, void *log_ctx)
 

Function Documentation

◆ get_datatype_size()

static int get_datatype_size ( DNNDataType  dt)
static

Definition at line 28 of file dnn_io_proc.c.

Referenced by ff_proc_from_dnn_to_frame(), and ff_proc_from_frame_to_dnn().

◆ ff_proc_from_dnn_to_frame()

int ff_proc_from_dnn_to_frame ( AVFrame frame,
DNNData output,
void *  log_ctx 
)

Definition at line 42 of file dnn_io_proc.c.

Referenced by infer_completion_callback().

◆ ff_proc_from_frame_to_dnn()

int ff_proc_from_frame_to_dnn ( AVFrame frame,
DNNData input,
void *  log_ctx 
)

Definition at line 182 of file dnn_io_proc.c.

Referenced by fill_model_input_ov(), fill_model_input_tf(), and fill_model_input_th().

◆ get_pixel_format()

static enum AVPixelFormat get_pixel_format ( DNNData data)
static

Definition at line 322 of file dnn_io_proc.c.

Referenced by ff_frame_to_dnn_classify(), and ff_frame_to_dnn_detect().

◆ ff_frame_to_dnn_classify()

int ff_frame_to_dnn_classify ( AVFrame frame,
DNNData input,
uint32_t  bbox_index,
void *  log_ctx 
)

Definition at line 340 of file dnn_io_proc.c.

Referenced by fill_model_input_ov().

◆ ff_frame_to_dnn_detect()

int ff_frame_to_dnn_detect ( AVFrame frame,
DNNData input,
void *  log_ctx 
)

Definition at line 423 of file dnn_io_proc.c.

Referenced by fill_model_input_ov(), and fill_model_input_tf().