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

avfilter/dnn_backend_tf: check TF_OperationOutputType return value

Browse Source 
This also fixed a warning: implicit conversion from enumeration
type 'TF_DataType' (aka 'enum TF_DataType') to different
enumeration type 'DNNDataType'.

Signed-off-by: Zhao Zhili <zhilizhao@tencent.com>
This commit is contained in:
Zhao Zhili 2023-04-30 23:39:00 +08:00
parent f3495ef4f8
commit d9f41a343e
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
libavfilter/dnn/dnn_backend_tf.c
View File

@ -274,6 +274,7 @@ static int get_input_tf(void *model, DNNData *input, const char *input_name)
TFModel *tf_model = model; TFModel *tf_model = model;
TFContext *ctx = &tf_model->ctx; TFContext *ctx = &tf_model->ctx;
TF_Status *status; TF_Status *status;
TF_DataType dt;
int64_t dims[4]; int64_t dims[4];
TF_Output tf_output; TF_Output tf_output;
@ -284,7 +285,18 @@ static int get_input_tf(void *model, DNNData *input, const char *input_name)
} }
tf_output.index = 0; tf_output.index = 0;
input->dt = TF_OperationOutputType(tf_output); dt = TF_OperationOutputType(tf_output);
switch (dt) {
case TF_FLOAT:
input->dt = DNN_FLOAT;
break;
case TF_UINT8:
input->dt = DNN_UINT8;
break;
default:
av_log(ctx, AV_LOG_ERROR, "Unsupported output type %d in model\n", dt);
return AVERROR(EINVAL);
}
input->order = DCO_RGB; input->order = DCO_RGB;
status = TF_NewStatus(); status = TF_NewStatus();