diff options
author | Ting Fu <ting.fu@intel.com> | 2021-05-06 16:46:09 +0800 |
---|---|---|
committer | Guo, Yejun <yejun.guo@intel.com> | 2021-05-11 10:28:35 +0800 |
commit | e42125edab9f6bbc3297334087608565218e119b (patch) | |
tree | da678f5f292981d595e42837f7c3f52e58f835b0 /libavfilter/vf_dnn_detect.c | |
parent | 1b1064054c8f3b4ad3b52d14f0c8ee1c4e8200fd (diff) |
lavfi/dnn_backend_tensorflow: support detect model
Signed-off-by: Ting Fu <ting.fu@intel.com>
Diffstat (limited to 'libavfilter/vf_dnn_detect.c')
-rw-r--r-- | libavfilter/vf_dnn_detect.c | 32 |
1 files changed, 31 insertions, 1 deletions
diff --git a/libavfilter/vf_dnn_detect.c b/libavfilter/vf_dnn_detect.c index 1dbe4f29a4..7d39acb653 100644 --- a/libavfilter/vf_dnn_detect.c +++ b/libavfilter/vf_dnn_detect.c @@ -203,10 +203,40 @@ static int read_detect_label_file(AVFilterContext *context) return 0; } +static int check_output_nb(DnnDetectContext *ctx, DNNBackendType backend_type, int output_nb) +{ + switch(backend_type) { + case DNN_TF: + if (output_nb != 4) { + av_log(ctx, AV_LOG_ERROR, "Only support tensorflow detect model with 4 outputs, \ + but get %d instead\n", output_nb); + return AVERROR(EINVAL); + } + return 0; + case DNN_OV: + if (output_nb != 1) { + av_log(ctx, AV_LOG_ERROR, "Dnn detect filter with openvino backend needs 1 output only, \ + but get %d instead\n", output_nb); + return AVERROR(EINVAL); + } + return 0; + default: + avpriv_report_missing_feature(ctx, "Dnn detect filter does not support current backend\n"); + return AVERROR(EINVAL); + } + return 0; +} + static av_cold int dnn_detect_init(AVFilterContext *context) { DnnDetectContext *ctx = context->priv; - int ret = ff_dnn_init(&ctx->dnnctx, DFT_ANALYTICS_DETECT, context); + DnnContext *dnn_ctx = &ctx->dnnctx; + int ret; + + ret = ff_dnn_init(&ctx->dnnctx, DFT_ANALYTICS_DETECT, context); + if (ret < 0) + return ret; + ret = check_output_nb(ctx, dnn_ctx->backend_type, dnn_ctx->nb_outputs); if (ret < 0) return ret; ff_dnn_set_detect_post_proc(&ctx->dnnctx, dnn_detect_post_proc); |