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>
This commit is contained in:
Guo, Yejun 2021-02-08 09:53:39 +08:00
parent 76fc6879e2
commit 0884063f88
1 changed files with 8 additions and 1 deletions

View File

@ -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{