dnn_interface: add interface to support async execution

Guo, Yejun committed 5 years ago

Commit 39f5cb41 parent 38089922 files · +13 −1Browse files

Unified · Split

2 files

@@ -33,7 +33,7 @@ DNNModule *ff_get_dnn_module(DNNBackendType backend_type) · 32 unmodified lines
33	{	33	{
34	DNNModule *dnn_module;	34	DNNModule *dnn_module;
35		35
36	dnn_module = av_malloc(sizeof(DNNModule));	36	dnn_module = av_mallocz(sizeof(DNNModule));
37	if(!dnn_module){	37	if(!dnn_module){
38	return NULL;	38	return NULL;
39	}	39	}

 typedef enum {DNN_FLOAT = 1, DNN_UINT8 = 4} DNNDataType;
+typedef enum {
+    DAST_FAIL,              // something wrong
+    DAST_EMPTY_QUEUE,       // no more inference result to get
+    DAST_NOT_READY,         // all queued inferences are not finished
+    DAST_SUCCESS            // got a result frame successfully
+} DNNAsyncStatusType;
 typedef struct DNNData{
     void *data;
     DNNDataType dt;
     // Executes model with specified input and output. Returns DNN_ERROR otherwise.
     DNNReturnType (*execute_model)(const DNNModel *model, const char *input_name, AVFrame *in_frame,
                                    const char **output_names, uint32_t nb_output, AVFrame *out_frame);
+    // Executes model with specified input and output asynchronously. Returns DNN_ERROR otherwise.
+    DNNReturnType (*execute_model_async)(const DNNModel *model, const char *input_name, AVFrame *in_frame,
+                                         const char **output_names, uint32_t nb_output, AVFrame *out_frame);
+    // Retrieve inference result.
+    DNNAsyncStatusType (*get_async_result)(const DNNModel *model, AVFrame **out);
     // Frees memory allocated for model.
     void (*free_model)(DNNModel **model);
 } DNNModule;

Keyboard shortcuts