mirror of https://git.ffmpeg.org/ffmpeg.git
lavfi/dnn_backend_openvino.c: only allow DFT_PROCESS_FRAME to get output dim
This commit is contained in:
parent
5106fe85f7
commit
d2ccbc966b
|
@ -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;
|
||||||
|
|
Loading…
Reference in New Issue