diff --git a/config.pbtxt b/config.pbtxt index 504259d..3396f89 100644 --- a/config.pbtxt +++ b/config.pbtxt @@ -7,10 +7,17 @@ input [ { name: "images" - optional: true + optional: false data_type: TYPE_FP32 dims: [ 1, 3, 640, 640 ] + }, + { + name: "text" + optional: true + data_type: TYPE_FP32 + dims: [ 2 ] + } ] @@ -19,7 +26,6 @@ output [ { name: "output0" - optional: true data_type: TYPE_FP32 dims: [ 1, 84, 8400 ]