diff options
| author | Ting Fu <ting.fu@intel.com> | 2021-05-06 16:46:08 +0800 |
|---|---|---|
| committer | Guo, Yejun <yejun.guo@intel.com> | 2021-05-11 10:28:35 +0800 |
| commit | 1b1064054c8f3b4ad3b52d14f0c8ee1c4e8200fd (patch) | |
| tree | fd7a4d3f397864a090ce5c4cd26c84fd2c64808f /libavfilter/dnn_filter_common.h | |
| parent | f02928eb5a75b2ee20dd94c30304f44a7d6f8de1 (diff) | |
lavfi/dnn_backend_tensorflow: add multiple outputs support
Signed-off-by: Ting Fu <ting.fu@intel.com>
Diffstat (limited to 'libavfilter/dnn_filter_common.h')
| -rw-r--r-- | libavfilter/dnn_filter_common.h | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/libavfilter/dnn_filter_common.h b/libavfilter/dnn_filter_common.h index e7736d2bac..09ddd8a5ca 100644 --- a/libavfilter/dnn_filter_common.h +++ b/libavfilter/dnn_filter_common.h @@ -30,10 +30,12 @@ typedef struct DnnContext { char *model_filename; DNNBackendType backend_type; char *model_inputname; - char *model_outputname; + char *model_outputnames_string; char *backend_options; int async; + char **model_outputnames; + uint32_t nb_outputs; DNNModule *dnn_module; DNNModel *model; } DnnContext; @@ -41,7 +43,7 @@ typedef struct DnnContext { #define DNN_COMMON_OPTIONS \ { "model", "path to model file", OFFSET(model_filename), AV_OPT_TYPE_STRING, { .str = NULL }, 0, 0, FLAGS },\ { "input", "input name of the model", OFFSET(model_inputname), AV_OPT_TYPE_STRING, { .str = NULL }, 0, 0, FLAGS },\ - { "output", "output name of the model", OFFSET(model_outputname), AV_OPT_TYPE_STRING, { .str = NULL }, 0, 0, FLAGS },\ + { "output", "output name of the model", OFFSET(model_outputnames_string), AV_OPT_TYPE_STRING, { .str = NULL }, 0, 0, FLAGS },\ { "backend_configs", "backend configs", OFFSET(backend_options), AV_OPT_TYPE_STRING, { .str = NULL }, 0, 0, FLAGS },\ { "options", "backend configs", OFFSET(backend_options), AV_OPT_TYPE_STRING, { .str = NULL }, 0, 0, FLAGS },\ { "async", "use DNN async inference", OFFSET(async), AV_OPT_TYPE_BOOL, { .i64 = 1}, 0, 1, FLAGS}, |