har0ke/FFmpeg
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

dnn_interface.h: add enum DNNColorOrder

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
libavfilter/dnn_interface.h
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 {DNN_FLOAT = 1, DNN_UINT8 = 4} DNNDataType;
typedef enum {
DCO_NONE,
DCO_BGR,
} DNNColorOrder;
typedef enum { typedef enum {
DAST_FAIL, // something wrong DAST_FAIL, // something wrong
DAST_EMPTY_QUEUE, // no more inference result to get DAST_EMPTY_QUEUE, // no more inference result to get
@ -52,8 +57,10 @@ typedef enum {
typedef struct DNNData{ typedef struct DNNData{
void *data; void *data;
DNNDataType dt;
int width, height, channels; int width, height, channels;
// dt and order together decide the color format
DNNDataType dt;
DNNColorOrder order;
} DNNData; } DNNData;
typedef struct DNNModel{ typedef struct DNNModel{