[FFmpeg-devel] [PATCH 3/8] lavfi/dnn_backend_tf: TFInferRequest Execution and Documentation
Shubhanshu Saxena
shubhanshu.e01 at gmail.com
Sat Jul 31 08:35:11 EEST 2021
This commit adds a function for execution of TFInferRequest and documentation
for functions related to TFInferRequest.
Signed-off-by: Shubhanshu Saxena <shubhanshu.e01 at gmail.com>
---
libavfilter/dnn/dnn_backend_tf.c | 45 ++++++++++++++++++++++++++++++++
1 file changed, 45 insertions(+)
diff --git a/libavfilter/dnn/dnn_backend_tf.c b/libavfilter/dnn/dnn_backend_tf.c
index 6443c2fd1d..805a0328b6 100644
--- a/libavfilter/dnn/dnn_backend_tf.c
+++ b/libavfilter/dnn/dnn_backend_tf.c
@@ -94,6 +94,13 @@ static void free_buffer(void *data, size_t length)
av_freep(&data);
}
+/**
+ * Free the contents of TensorFlow inference request.
+ * It does not free the TFInferRequest instance.
+ *
+ * @param request pointer to TFInferRequest instance.
+ * NULL pointer is allowed.
+ */
static void tf_free_request(TFInferRequest *request)
{
if (!request)
@@ -116,6 +123,12 @@ static void tf_free_request(TFInferRequest *request)
}
}
+/**
+ * Create a TensorFlow inference request. All properties
+ * are initially unallocated and set as NULL.
+ *
+ * @return pointer to the allocated TFInferRequest instance.
+ */
static TFInferRequest *tf_create_inference_request(void)
{
TFInferRequest *infer_request = av_malloc(sizeof(TFInferRequest));
@@ -126,6 +139,38 @@ static TFInferRequest *tf_create_inference_request(void)
return infer_request;
}
+/**
+ * Start synchronous inference for the TensorFlow model.
+ *
+ * @param request pointer to the TFRequestItem for inference
+ * @retval DNN_SUCCESS if execution is successful
+ * @retval DNN_ERROR if execution fails
+ */
+static DNNReturnType tf_start_inference(void *args)
+{
+ TFRequestItem *request = args;
+ TFInferRequest *infer_request = request->infer_request;
+ InferenceItem *inference = request->inference;
+ TaskItem *task = inference->task;
+ TFModel *tf_model = task->model;
+
+ if (!request) {
+ av_log(&tf_model->ctx, AV_LOG_ERROR, "TFRequestItem is NULL\n");
+ return DNN_ERROR;
+ }
+
+ TF_SessionRun(tf_model->session, NULL,
+ infer_request->tf_input, &infer_request->input_tensor, 1,
+ infer_request->tf_outputs, infer_request->output_tensors,
+ task->nb_output, NULL, 0, NULL,
+ tf_model->status);
+ if (TF_GetCode(tf_model->status) != TF_OK) {
+ av_log(&tf_model->ctx, AV_LOG_ERROR, "%s", TF_Message(tf_model->status));
+ return DNN_ERROR;
+ }
+ return DNN_SUCCESS;
+}
+
static DNNReturnType extract_inference_from_task(TaskItem *task, Queue *inference_queue)
{
TFModel *tf_model = task->model;
--
2.25.1
More information about the ffmpeg-devel
mailing list