summaryrefslogtreecommitdiff
path: root/libavfilter/dnn_interface.h
diff options
context:
space:
mode:
authorGuo, Yejun <yejun.guo@intel.com>2021-02-08 09:53:39 +0800
committerGuo, Yejun <yejun.guo@intel.com>2021-02-18 09:59:37 +0800
commit0884063f88a9b726b296722cabe397b12337b16a (patch)
tree5d19839b9f2a8f52b9cf360daac82b9df92034af /libavfilter/dnn_interface.h
parent76fc6879e2c59929e7d81a72986bb4950053177e (diff)
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 <yejun.guo@intel.com>
Diffstat (limited to 'libavfilter/dnn_interface.h')
-rw-r--r--libavfilter/dnn_interface.h9
1 files changed, 8 insertions, 1 deletions
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
@@ -37,6 +37,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
DAST_NOT_READY, // all queued inferences are not finished
@@ -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{