diff options
author | Guo, Yejun <yejun.guo@intel.com> | 2021-03-15 11:42:27 +0300 |
---|---|---|
committer | Guo, Yejun <yejun.guo@intel.com> | 2021-04-08 04:23:02 +0300 |
commit | d2ccbc966b151f2500aa43e789288af188258ba3 (patch) | |
tree | 5d2d2605742d2bf50f4ff566c41462ee1768e658 /libavfilter | |
parent | 5106fe85f71459f0f293019e682edec0c3c26cee (diff) |
lavfi/dnn_backend_openvino.c: only allow DFT_PROCESS_FRAME to get output dim
Diffstat (limited to 'libavfilter')
-rw-r--r-- | libavfilter/dnn/dnn_backend_openvino.c | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/libavfilter/dnn/dnn_backend_openvino.c b/libavfilter/dnn/dnn_backend_openvino.c index 9a47d74c15..66845fbd51 100644 --- a/libavfilter/dnn/dnn_backend_openvino.c +++ b/libavfilter/dnn/dnn_backend_openvino.c @@ -493,6 +493,11 @@ static DNNReturnType get_output_ov(void *model, const char *input_name, int inpu IEStatusCode status; input_shapes_t input_shapes; + if (ov_model->model->func_type != DFT_PROCESS_FRAME) { + av_log(ctx, AV_LOG_ERROR, "Get output dim only when processing frame.\n"); + return DNN_ERROR; + } + if (ctx->options.input_resizable) { status = ie_network_get_input_shapes(ov_model->network, &input_shapes); input_shapes.shapes->shape.dims[2] = input_height; |