vf_dnn_detect.c File Reference

#include "libavformat/avio.h"
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "libavutil/avassert.h"
#include "libavutil/imgutils.h"
#include "filters.h"
#include "dnn_filter_common.h"
#include "formats.h"
#include "internal.h"
#include "libavutil/time.h"
#include "libavutil/avstring.h"
#include "libavutil/detection_bbox.h"

Go to the source code of this file.

Data Structures
struct	DnnDetectContext

Macros
#define	OFFSET(x)   offsetof(DnnDetectContext, dnnctx.x)
#define	OFFSET2(x)   offsetof(DnnDetectContext, x)
#define	FLAGS   AV_OPT_FLAG_FILTERING_PARAM | AV_OPT_FLAG_VIDEO_PARAM

Functions
	AVFILTER_DEFINE_CLASS (dnn_detect)
static int	dnn_detect_post_proc_ov (AVFrame *frame, DNNData *output, AVFilterContext *filter_ctx)
static int	dnn_detect_post_proc_tf (AVFrame *frame, DNNData *output, AVFilterContext *filter_ctx)
static int	dnn_detect_post_proc (AVFrame *frame, DNNData *output, uint32_t nb, AVFilterContext *filter_ctx)
static void	free_detect_labels (DnnDetectContext *ctx)
static int	read_detect_label_file (AVFilterContext *context)
static int	check_output_nb (DnnDetectContext *ctx, DNNBackendType backend_type, int output_nb)
static av_cold int	dnn_detect_init (AVFilterContext *context)
static int	dnn_detect_flush_frame (AVFilterLink *outlink, int64_t pts, int64_t *out_pts)
static int	dnn_detect_activate (AVFilterContext *filter_ctx)
static av_cold void	dnn_detect_uninit (AVFilterContext *context)

Variables
static const AVOption	dnn_detect_options []
static enum AVPixelFormat	pix_fmts []
static const AVFilterPad	dnn_detect_inputs []
static const AVFilterPad	dnn_detect_outputs []
const AVFilter	ff_vf_dnn_detect

Detailed Description

implementing an object detecting filter using deep learning networks.

Definition in file vf_dnn_detect.c.

Macro Definition Documentation

OFFSET

#define OFFSET

(

x

)

offsetof(DnnDetectContext, dnnctx.x)

Definition at line 46 of file vf_dnn_detect.c.

OFFSET2

#define OFFSET2

(

x

)

offsetof(DnnDetectContext, x)

Definition at line 47 of file vf_dnn_detect.c.

FLAGS

#define FLAGS   AV_OPT_FLAG_FILTERING_PARAM | AV_OPT_FLAG_VIDEO_PARAM

Definition at line 48 of file vf_dnn_detect.c.

Function Documentation

AVFILTER_DEFINE_CLASS()

AVFILTER_DEFINE_CLASS

(

dnn_detect

)

dnn_detect_post_proc_ov()

static int dnn_detect_post_proc_ov ( AVFrame *  frame, DNNData *  output, AVFilterContext *  filter_ctx  )

static

Definition at line 65 of file vf_dnn_detect.c.

Referenced by dnn_detect_post_proc().

dnn_detect_post_proc_tf()

static int dnn_detect_post_proc_tf ( AVFrame *  frame, DNNData *  output, AVFilterContext *  filter_ctx  )

static

Definition at line 142 of file vf_dnn_detect.c.

Referenced by dnn_detect_post_proc().

dnn_detect_post_proc()

static int dnn_detect_post_proc ( AVFrame *  frame, DNNData *  output, uint32_t  nb, AVFilterContext *  filter_ctx  )

static

Definition at line 217 of file vf_dnn_detect.c.

Referenced by dnn_detect_init().

free_detect_labels()

static void free_detect_labels ( DnnDetectContext *  ctx )

static

Definition at line 232 of file vf_dnn_detect.c.

Referenced by dnn_detect_uninit().

read_detect_label_file()

static int read_detect_label_file ( AVFilterContext *  context )

static

Definition at line 241 of file vf_dnn_detect.c.

Referenced by dnn_detect_init().

check_output_nb()

static int check_output_nb ( DnnDetectContext *  ctx, DNNBackendType  backend_type, int  output_nb  )

static

Definition at line 299 of file vf_dnn_detect.c.

Referenced by dnn_detect_init().

dnn_detect_init()

static av_cold int dnn_detect_init ( AVFilterContext *  context )

static

Definition at line 323 of file vf_dnn_detect.c.

dnn_detect_flush_frame()

static int dnn_detect_flush_frame ( AVFilterLink *  outlink, int64_t  pts, int64_t *  out_pts  )

static

Definition at line 352 of file vf_dnn_detect.c.

Referenced by dnn_detect_activate().

dnn_detect_activate()

static int dnn_detect_activate ( AVFilterContext *  filter_ctx )

static

Definition at line 380 of file vf_dnn_detect.c.

dnn_detect_uninit()

static av_cold void dnn_detect_uninit ( AVFilterContext *  context )

static

Definition at line 436 of file vf_dnn_detect.c.

Variable Documentation

dnn_detect_options

const AVOption dnn_detect_options[]

static

Initial value:

= {
    { "dnn_backend", "DNN backend",                OFFSET(backend_type),     AV_OPT_TYPE_INT,       { .i64 = 2 },    INT_MIN, INT_MAX, FLAGS, "backend" },
 
 
 
 
 
 
    DNN_COMMON_OPTIONS
    { "confidence",  "threshold of confidence",    OFFSET2(confidence),      AV_OPT_TYPE_FLOAT,     { .dbl = 0.5 },  0, 1, FLAGS},
    { "labels",      "path to labels file",        OFFSET2(labels_filename), AV_OPT_TYPE_STRING,    { .str = NULL }, 0, 0, FLAGS },
    { NULL }
}

Definition at line 49 of file vf_dnn_detect.c.

pix_fmts

enum AVPixelFormat pix_fmts[]

static

Initial value:

= {
    AV_PIX_FMT_RGB24, AV_PIX_FMT_BGR24,
    AV_PIX_FMT_GRAY8, AV_PIX_FMT_GRAYF32,
    AV_PIX_FMT_YUV420P, AV_PIX_FMT_YUV422P,
    AV_PIX_FMT_YUV444P, AV_PIX_FMT_YUV410P, AV_PIX_FMT_YUV411P,
    AV_PIX_FMT_NV12,
    AV_PIX_FMT_NONE
}

Definition at line 343 of file vf_dnn_detect.c.

dnn_detect_inputs

const AVFilterPad dnn_detect_inputs[]

static

Initial value:

= {
    {
        .name         = "default",
        .type         = AVMEDIA_TYPE_VIDEO,
    },
}

Definition at line 443 of file vf_dnn_detect.c.

dnn_detect_outputs

const AVFilterPad dnn_detect_outputs[]

static

Initial value:

= {
    {
        .name = "default",
        .type = AVMEDIA_TYPE_VIDEO,
    },
}

Definition at line 450 of file vf_dnn_detect.c.

ff_vf_dnn_detect

const AVFilter ff_vf_dnn_detect

Initial value:

= {
    .name          = "dnn_detect",
    .description   = NULL_IF_CONFIG_SMALL("Apply DNN detect filter to the input."),
    .priv_size     = sizeof(DnnDetectContext),
    .init          = dnn_detect_init,
    .uninit        = dnn_detect_uninit,
    FILTER_INPUTS(dnn_detect_inputs),
    FILTER_OUTPUTS(dnn_detect_outputs),
    FILTER_PIXFMTS_ARRAY(pix_fmts),
    .priv_class    = &dnn_detect_class,
    .activate      = dnn_detect_activate,
}

Definition at line 457 of file vf_dnn_detect.c.