dnn_backend_tf.c File Reference

#include "dnn_backend_tf.h"
#include "dnn_backend_native.h"
#include "dnn_backend_native_layer_conv2d.h"
#include "dnn_backend_native_layer_depth2space.h"
#include "libavformat/avio.h"
#include "libavutil/avassert.h"
#include "../internal.h"
#include "dnn_backend_native_layer_pad.h"
#include "dnn_backend_native_layer_maximum.h"
#include "dnn_io_proc.h"
#include <tensorflow/c/c_api.h>

Go to the source code of this file.

Data Structures
struct	TFOptions
struct	TFContext
struct	TFModel

Macros
#define	OFFSET(x)   offsetof(TFContext, x)
#define	FLAGS   AV_OPT_FLAG_FILTERING_PARAM
#define	NAME_BUFFER_SIZE   256
#define	CLEANUP_ON_ERROR(tf_model)

Functions
	AVFILTER_DEFINE_CLASS (dnn_tensorflow)
static DNNReturnType	execute_model_tf (const DNNModel *model, const char *input_name, AVFrame *in_frame, const char **output_names, uint32_t nb_output, AVFrame *out_frame, int do_ioproc)
static void	free_buffer (void *data, size_t length)
static TF_Buffer *	read_graph (const char *model_filename)
static TF_Tensor *	allocate_input_tensor (const DNNData *input)
static DNNReturnType	get_input_tf (void *model, DNNData *input, const char *input_name)
static DNNReturnType	get_output_tf (void *model, const char *input_name, int input_width, int input_height, const char *output_name, int *output_width, int *output_height)
static DNNReturnType	load_tf_model (TFModel *tf_model, const char *model_filename)
static DNNReturnType	add_conv_layer (TFModel *tf_model, TF_Operation *transpose_op, TF_Operation **cur_op, ConvolutionalParams *params, const int layer)
static DNNReturnType	add_depth_to_space_layer (TFModel *tf_model, TF_Operation **cur_op, DepthToSpaceParams *params, const int layer)
static DNNReturnType	add_pad_layer (TFModel *tf_model, TF_Operation **cur_op, LayerPadParams *params, const int layer)
static DNNReturnType	add_maximum_layer (TFModel *tf_model, TF_Operation **cur_op, DnnLayerMaximumParams *params, const int layer)
static DNNReturnType	load_native_model (TFModel *tf_model, const char *model_filename)
DNNModel *	ff_dnn_load_model_tf (const char *model_filename, DNNFunctionType func_type, const char *options, AVFilterContext *filter_ctx)
DNNReturnType	ff_dnn_execute_model_tf (const DNNModel *model, const char *input_name, AVFrame *in_frame, const char **output_names, uint32_t nb_output, AVFrame *out_frame)
void	ff_dnn_free_model_tf (DNNModel **model)

Variables
static const AVOption	dnn_tensorflow_options []

Detailed Description

DNN tensorflow backend implementation.

Definition in file dnn_backend_tf.c.

Macro Definition Documentation

OFFSET

#define OFFSET

(

x

)

offsetof(TFContext, x)

Definition at line 56 of file dnn_backend_tf.c.

FLAGS

#define FLAGS   AV_OPT_FLAG_FILTERING_PARAM

Definition at line 57 of file dnn_backend_tf.c.

NAME_BUFFER_SIZE

#define NAME_BUFFER_SIZE   256

Definition at line 315 of file dnn_backend_tf.c.

CLEANUP_ON_ERROR

#define CLEANUP_ON_ERROR

(

tf_model

)

Value:

    { \
        TF_DeleteGraph(tf_model->graph); \
        TF_DeleteStatus(tf_model->status); \
        av_log(ctx, AV_LOG_ERROR, "Failed to set value or add operator to layer\n"); \
        return DNN_ERROR; \
    }

Function Documentation

AVFILTER_DEFINE_CLASS()

AVFILTER_DEFINE_CLASS

(

dnn_tensorflow

)

execute_model_tf()

static DNNReturnType execute_model_tf ( const DNNModel *  model, const char *  input_name, AVFrame *  in_frame, const char **  output_names, uint32_t  nb_output, AVFrame *  out_frame, int  do_ioproc  )

static

Definition at line 713 of file dnn_backend_tf.c.

Referenced by ff_dnn_execute_model_tf(), and get_output_tf().

free_buffer()

static void free_buffer ( void *  data, size_t  length  )

static

Definition at line 69 of file dnn_backend_tf.c.

Referenced by read_graph().

read_graph()

static TF_Buffer* read_graph ( const char *  model_filename )

static

Definition at line 74 of file dnn_backend_tf.c.

Referenced by load_tf_model().

allocate_input_tensor()

static TF_Tensor* allocate_input_tensor ( const DNNData *  input )

static

Definition at line 107 of file dnn_backend_tf.c.

Referenced by execute_model_tf().

get_input_tf()

static DNNReturnType get_input_tf ( void *  model, DNNData *  input, const char *  input_name  )

static

Definition at line 129 of file dnn_backend_tf.c.

Referenced by execute_model_tf(), and ff_dnn_load_model_tf().

get_output_tf()

static DNNReturnType get_output_tf ( void *  model, const char *  input_name, int  input_width, int  input_height, const char *  output_name, int *  output_width, int *  output_height  )

static

Definition at line 164 of file dnn_backend_tf.c.

Referenced by ff_dnn_load_model_tf().

load_tf_model()

static DNNReturnType load_tf_model ( TFModel *  tf_model, const char *  model_filename  )

static

Definition at line 197 of file dnn_backend_tf.c.

Referenced by ff_dnn_load_model_tf().

add_conv_layer()

static DNNReturnType add_conv_layer ( TFModel *  tf_model, TF_Operation *  transpose_op, TF_Operation **  cur_op, ConvolutionalParams *  params, const int  layer  )

static

Definition at line 317 of file dnn_backend_tf.c.

Referenced by load_native_model().

add_depth_to_space_layer()

static DNNReturnType add_depth_to_space_layer ( TFModel *  tf_model, TF_Operation **  cur_op, DepthToSpaceParams *  params, const int  layer  )

static

Definition at line 442 of file dnn_backend_tf.c.

Referenced by load_native_model().

add_pad_layer()

static DNNReturnType add_pad_layer ( TFModel *  tf_model, TF_Operation **  cur_op, LayerPadParams *  params, const int  layer  )

static

Definition at line 466 of file dnn_backend_tf.c.

Referenced by load_native_model().

add_maximum_layer()

static DNNReturnType add_maximum_layer ( TFModel *  tf_model, TF_Operation **  cur_op, DnnLayerMaximumParams *  params, const int  layer  )

static

Definition at line 521 of file dnn_backend_tf.c.

Referenced by load_native_model().

load_native_model()

static DNNReturnType load_native_model ( TFModel *  tf_model, const char *  model_filename  )

static

Definition at line 567 of file dnn_backend_tf.c.

Referenced by ff_dnn_load_model_tf().

ff_dnn_load_model_tf()

DNNModel* ff_dnn_load_model_tf	(	const char *	model_filename,
		DNNFunctionType	func_type,
		const char *	options,
		AVFilterContext *	filter_ctx
	)

Definition at line 667 of file dnn_backend_tf.c.

Referenced by ff_get_dnn_module().

ff_dnn_execute_model_tf()

DNNReturnType ff_dnn_execute_model_tf	(	const DNNModel *	model,
		const char *	input_name,
		AVFrame *	in_frame,
		const char **	output_names,
		uint32_t	nb_output,
		AVFrame *	out_frame
	)

Definition at line 823 of file dnn_backend_tf.c.

Referenced by ff_get_dnn_module().

ff_dnn_free_model_tf()

void ff_dnn_free_model_tf

(

DNNModel **

model

)

Definition at line 842 of file dnn_backend_tf.c.

Referenced by ff_get_dnn_module().

Variable Documentation

dnn_tensorflow_options

const AVOption dnn_tensorflow_options[]

static

Initial value:

= {
    { "sess_config", "config for SessionOptions", OFFSET(options.sess_config), AV_OPT_TYPE_STRING, { .str = NULL }, 0, 0, FLAGS },
    { NULL }
}

Definition at line 58 of file dnn_backend_tf.c.