[FFmpeg-devel] [PATCH 05/10] dnn_interface.h: add enum DNNColorOrder
Guo, Yejun
yejun.guo at intel.com
Wed Feb 10 06:58:26 EET 2021
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 at intel.com>
---
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{
--
2.17.1
More information about the ffmpeg-devel
mailing list