highperfocused/ffmpeg
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

lavfi/dnn_backend_openvino.c: only allow DFT_PROCESS_FRAME to get output dim

Browse Source
This commit is contained in:
Guo, Yejun
2021-03-15 16:42:27 +08:00
parent 5106fe85f7
commit d2ccbc966b
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
libavfilter/dnn/dnn_backend_openvino.c
View File

@@ -493,6 +493,11 @@ static DNNReturnType get_output_ov(void *model, const char *input_name, int inpu
IEStatusCode status; IEStatusCode status;
input_shapes_t input_shapes; 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) { if (ctx->options.input_resizable) {
status = ie_network_get_input_shapes(ov_model->network, &input_shapes); status = ie_network_get_input_shapes(ov_model->network, &input_shapes);
input_shapes.shapes->shape.dims[2] = input_height; input_shapes.shapes->shape.dims[2] = input_height;