From 0884063f88a9b726b296722cabe397b12337b16a Mon Sep 17 00:00:00 2001 From: "Guo, Yejun" Date: Mon, 8 Feb 2021 09:53:39 +0800 Subject: [PATCH] dnn_interface.h: add enum DNNColorOrder the data type and order together decide the color format, we could not use AVPixelFormat directly because not all the possible formats are covered by it. Signed-off-by: Guo, Yejun --- libavfilter/dnn_interface.h | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/libavfilter/dnn_interface.h b/libavfilter/dnn_interface.h index 2fb9b15676..d3a0c58a61 100644 --- a/libavfilter/dnn_interface.h +++ b/libavfilter/dnn_interface.h @@ -36,6 +36,11 @@ typedef enum {DNN_NATIVE, DNN_TF, DNN_OV} DNNBackendType; typedef enum {DNN_FLOAT = 1, DNN_UINT8 = 4} DNNDataType; +typedef enum { + DCO_NONE, + DCO_BGR, +} DNNColorOrder; + typedef enum { DAST_FAIL, // something wrong DAST_EMPTY_QUEUE, // no more inference result to get @@ -52,8 +57,10 @@ typedef enum { typedef struct DNNData{ void *data; - DNNDataType dt; int width, height, channels; + // dt and order together decide the color format + DNNDataType dt; + DNNColorOrder order; } DNNData; typedef struct DNNModel{