vf_derain.c File Reference

#include "libavformat/avio.h"
#include "libavutil/opt.h"
#include "avfilter.h"
#include "dnn_filter_common.h"
#include "formats.h"
#include "internal.h"

Go to the source code of this file.

Data Structures
struct	DRContext

Macros
#define	OFFSET(x)   offsetof(DRContext, x)
#define	FLAGS   AV_OPT_FLAG_FILTERING_PARAM | AV_OPT_FLAG_VIDEO_PARAM

Functions
	AVFILTER_DEFINE_CLASS (derain)
static int	filter_frame (AVFilterLink *inlink, AVFrame *in)
static av_cold int	init (AVFilterContext *ctx)
static av_cold void	uninit (AVFilterContext *ctx)

Variables
static const AVOption	derain_options []
static const AVFilterPad	derain_inputs []
static const AVFilterPad	derain_outputs []
const AVFilter	ff_vf_derain

Detailed Description

Filter implementing image derain filter using deep convolutional networks. http://openaccess.thecvf.com/content_ECCV_2018/html/Xia_Li_Recurrent_Squeeze-and-Excitation_Context_ECCV_2018_paper.html

Definition in file vf_derain.c.

Macro Definition Documentation

OFFSET

#define OFFSET

(

x

)

offsetof(DRContext, x)

Definition at line 40 of file vf_derain.c.

FLAGS

#define FLAGS   AV_OPT_FLAG_FILTERING_PARAM | AV_OPT_FLAG_VIDEO_PARAM

Definition at line 41 of file vf_derain.c.

Function Documentation

AVFILTER_DEFINE_CLASS()

AVFILTER_DEFINE_CLASS

(

derain

)

filter_frame()

static int filter_frame ( AVFilterLink *  inlink, AVFrame *  in  )

static

Definition at line 59 of file vf_derain.c.

init()

static av_cold int init ( AVFilterContext *  ctx )

static

Definition at line 94 of file vf_derain.c.

uninit()

static av_cold void uninit ( AVFilterContext *  ctx )

static

Definition at line 100 of file vf_derain.c.

Variable Documentation

derain_options

const AVOption derain_options[]

static

Initial value:

= {
    { "filter_type", "filter type(derain/dehaze)",  OFFSET(filter_type),    AV_OPT_TYPE_INT,    { .i64 = 0 },    0, 1, FLAGS, "type" },
    { "derain",      "derain filter flag",          0,                      AV_OPT_TYPE_CONST,  { .i64 = 0 },    0, 0, FLAGS, "type" },
    { "dehaze",      "dehaze filter flag",          0,                      AV_OPT_TYPE_CONST,  { .i64 = 1 },    0, 0, FLAGS, "type" },
    { "dnn_backend", "DNN backend",                 OFFSET(dnnctx.backend_type),   AV_OPT_TYPE_INT,    { .i64 = 0 },    0, 1, FLAGS, "backend" },
    { "native",      "native backend flag",         0,                      AV_OPT_TYPE_CONST,  { .i64 = 0 },    0, 0, FLAGS, "backend" },
 
 
 
    { "model",       "path to model file",          OFFSET(dnnctx.model_filename),   AV_OPT_TYPE_STRING,    { .str = NULL }, 0, 0, FLAGS },
    { "input",       "input name of the model",     OFFSET(dnnctx.model_inputname),  AV_OPT_TYPE_STRING,    { .str = "x" },  0, 0, FLAGS },
    { "output",      "output name of the model",    OFFSET(dnnctx.model_outputnames_string), AV_OPT_TYPE_STRING,    { .str = "y" },  0, 0, FLAGS },
    { NULL }
}

Definition at line 42 of file vf_derain.c.

derain_inputs

const AVFilterPad derain_inputs[]

static

Initial value:

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

Definition at line 106 of file vf_derain.c.

derain_outputs

const AVFilterPad derain_outputs[]

static

Initial value:

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

Definition at line 114 of file vf_derain.c.

ff_vf_derain

const AVFilter ff_vf_derain

Initial value:

= {
    .name          = "derain",
    .description   = NULL_IF_CONFIG_SMALL("Apply derain filter to the input."),
    .priv_size     = sizeof(DRContext),
    .init          = init,
    .uninit        = uninit,
    FILTER_INPUTS(derain_inputs),
    FILTER_OUTPUTS(derain_outputs),
    FILTER_SINGLE_PIXFMT(AV_PIX_FMT_RGB24),
    .priv_class    = &derain_class,
    .flags         = AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC,
}

Definition at line 121 of file vf_derain.c.