vf_sr.c File Reference

Filter implementing image super-resolution using deep convolutional networks. More...

#include "avfilter.h"
#include "formats.h"
#include "internal.h"
#include "libavutil/opt.h"
#include "libavformat/avio.h"
#include "libswscale/swscale.h"
#include "dnn_interface.h"

Go to the source code of this file.

Data Structures
struct	SRContext

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

Functions
	AVFILTER_DEFINE_CLASS (sr)
static av_cold int	init (AVFilterContext *context)
static int	query_formats (AVFilterContext *context)
static int	config_props (AVFilterLink *inlink)
static int	filter_frame (AVFilterLink *inlink, AVFrame *in)
static av_cold void	uninit (AVFilterContext *context)

Variables
static const AVOption	sr_options []
static const AVFilterPad	sr_inputs []
static const AVFilterPad	sr_outputs []
AVFilter	ff_vf_sr

Detailed Description

Filter implementing image super-resolution using deep convolutional networks.

https://arxiv.org/abs/1501.00092 https://arxiv.org/abs/1609.05158

Definition in file vf_sr.c.

Macro Definition Documentation

#define OFFSET

(

x

)

offsetof(SRContext, x)

Definition at line 49 of file vf_sr.c.

#define FLAGS   AV_OPT_FLAG_FILTERING_PARAM | AV_OPT_FLAG_VIDEO_PARAM

Definition at line 50 of file vf_sr.c.

Function Documentation

AVFILTER_DEFINE_CLASS

(

sr

)

static av_cold int init ( AVFilterContext *  context )

static

Definition at line 64 of file vf_sr.c.

static int query_formats ( AVFilterContext *  context )

static

Definition at line 97 of file vf_sr.c.

static int config_props ( AVFilterLink *  inlink )

static

Definition at line 113 of file vf_sr.c.

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

static

Definition at line 214 of file vf_sr.c.

static av_cold void uninit ( AVFilterContext *  context )

static

Definition at line 265 of file vf_sr.c.

Variable Documentation

const AVOption sr_options[]

static

Initial value:

= {
    { "dnn_backend", "DNN backend used for model execution", OFFSET(backend_type), AV_OPT_TYPE_FLAGS, { .i64 = 0 }, 0, 1, FLAGS, "backend" },
    { "native", "native backend flag", 0, AV_OPT_TYPE_CONST, { .i64 = 0 }, 0, 0, FLAGS, "backend" },



    { "scale_factor", "scale factor for SRCNN model", OFFSET(scale_factor), AV_OPT_TYPE_INT, { .i64 = 2 }, 2, 4, FLAGS },
    { "model", "path to model file specifying network architecture and its parameters", OFFSET(model_filename), AV_OPT_TYPE_STRING, {.str=NULL}, 0, 0, FLAGS },
    { NULL }
}

Definition at line 51 of file vf_sr.c.

const AVFilterPad sr_inputs[]

static

Initial value:

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

Definition at line 282 of file vf_sr.c.

const AVFilterPad sr_outputs[]

static

Initial value:

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

Definition at line 292 of file vf_sr.c.

AVFilter ff_vf_sr

Initial value:

= {
    .name          = "sr",
    .description   = NULL_IF_CONFIG_SMALL("Apply DNN-based image super resolution to the input."),
    .priv_size     = sizeof(SRContext),
    .init          = init,
    .uninit        = uninit,
    .query_formats = query_formats,
    .inputs        = sr_inputs,
    .outputs       = sr_outputs,
    .priv_class    = &sr_class,
    .flags         = AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC | AVFILTER_FLAG_SLICE_THREADS,
}

Definition at line 300 of file vf_sr.c.