FFmpeg
Data Structures | Macros | Functions | Variables
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

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",
.filter_frame = filter_frame,
},
}

Definition at line 106 of file vf_derain.c.

◆ derain_outputs

const AVFilterPad derain_outputs[]
static
Initial value:
= {
{
.name = "default",
},
}

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,
.priv_class = &derain_class,
}

Definition at line 121 of file vf_derain.c.

DRContext
Definition: vf_derain.c:34
init
static av_cold int init(AVFilterContext *ctx)
Definition: vf_derain.c:94
derain_inputs
static const AVFilterPad derain_inputs[]
Definition: vf_derain.c:106
FILTER_INPUTS
#define FILTER_INPUTS(array)
Definition: internal.h:191
NULL
#define NULL
Definition: coverity.c:32
derain_outputs
static const AVFilterPad derain_outputs[]
Definition: vf_derain.c:114
filter_frame
static int filter_frame(AVFilterLink *inlink, AVFrame *in)
Definition: vf_derain.c:59
FLAGS
#define FLAGS
Definition: vf_derain.c:41
AV_PIX_FMT_RGB24
@ AV_PIX_FMT_RGB24
packed RGB 8:8:8, 24bpp, RGBRGB...
Definition: pixfmt.h:68
NULL_IF_CONFIG_SMALL
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
Definition: internal.h:117
AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC
#define AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC
Some filters support a generic "enable" expression option that can be used to enable or disable a fil...
Definition: avfilter.h:146
FILTER_SINGLE_PIXFMT
#define FILTER_SINGLE_PIXFMT(pix_fmt_)
Definition: internal.h:181
AV_OPT_TYPE_INT
@ AV_OPT_TYPE_INT
Definition: opt.h:224
AVMEDIA_TYPE_VIDEO
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
OFFSET
#define OFFSET(x)
Definition: vf_derain.c:40
FILTER_OUTPUTS
#define FILTER_OUTPUTS(array)
Definition: internal.h:192
uninit
static av_cold void uninit(AVFilterContext *ctx)
Definition: vf_derain.c:100
AV_OPT_TYPE_STRING
@ AV_OPT_TYPE_STRING
Definition: opt.h:228
AV_OPT_TYPE_CONST
@ AV_OPT_TYPE_CONST
Definition: opt.h:233