vf_sr.c File Reference

#include "avfilter.h"
#include "formats.h"
#include "internal.h"
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "libavformat/avio.h"
#include "libswscale/swscale.h"
#include "dnn_filter_common.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	config_output (AVFilterLink *outlink)
static int	filter_frame (AVFilterLink *inlink, AVFrame *in)
static av_cold void	uninit (AVFilterContext *context)

Variables
static const AVOption	sr_options []
static enum AVPixelFormat	pixel_formats []
static const AVFilterPad	sr_inputs []
static const AVFilterPad	sr_outputs []
const 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

OFFSET

#define OFFSET

(

x

)

offsetof(SRContext, x)

Definition at line 46 of file vf_sr.c.

FLAGS

#define FLAGS   AV_OPT_FLAG_FILTERING_PARAM | AV_OPT_FLAG_VIDEO_PARAM

Definition at line 47 of file vf_sr.c.

Function Documentation

AVFILTER_DEFINE_CLASS()

AVFILTER_DEFINE_CLASS

(

sr

)

init()

static av_cold int init ( AVFilterContext *  context )

static

Definition at line 63 of file vf_sr.c.

config_output()

static int config_output ( AVFilterLink *  outlink )

static

Definition at line 75 of file vf_sr.c.

filter_frame()

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

static

Definition at line 117 of file vf_sr.c.

uninit()

static av_cold void uninit ( AVFilterContext *  context )

static

Definition at line 167 of file vf_sr.c.

Variable Documentation

sr_options

const AVOption sr_options[]

static

Initial value:

= {
    { "dnn_backend", "DNN backend used for model execution", 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" },
 
 
 
    { "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(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 48 of file vf_sr.c.

pixel_formats

enum AVPixelFormat pixel_formats[]

static

Initial value:

= {
    AV_PIX_FMT_YUV420P, AV_PIX_FMT_YUV422P, AV_PIX_FMT_YUV444P,
    AV_PIX_FMT_YUV410P, AV_PIX_FMT_YUV411P, AV_PIX_FMT_GRAY8,
    AV_PIX_FMT_NONE
}

Definition at line 69 of file vf_sr.c.

Referenced by main().

sr_inputs

const AVFilterPad sr_inputs[]

static

Initial value:

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

Definition at line 176 of file vf_sr.c.

sr_outputs

const AVFilterPad sr_outputs[]

static

Initial value:

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

Definition at line 184 of file vf_sr.c.

ff_vf_sr

const 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,
    FILTER_INPUTS(sr_inputs),
    FILTER_OUTPUTS(sr_outputs),
    FILTER_PIXFMTS_ARRAY(pixel_formats),
    .priv_class    = &sr_class,
}

Definition at line 192 of file vf_sr.c.