RepoMirrors
/
ffmpeg
mirror of https://git.ffmpeg.org/ffmpeg.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

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