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)
static void	detect_write_tensor (DNNData *input, const uint8_t *src, int src_linesize, int w, int h)
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(), infer_completion_callback(), infer_completion_callback(), and 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_onnx(), 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().

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().

detect_write_tensor()

static void detect_write_tensor ( DNNData * input, const uint8_t * src, int src_linesize, int w, int h )

static

Definition at line 426 of file dnn_io_proc.c.

Referenced by ff_frame_to_dnn_detect().

ff_frame_to_dnn_detect()

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

Definition at line 461 of file dnn_io_proc.c.

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