diff --git a/config.pbtxt b/config.pbtxt index 33be1a2..16853b9 100644 --- a/config.pbtxt +++ b/config.pbtxt @@ -4,30 +4,21 @@ platform: "onnxruntime_onnx" max_batch_size: 0 input [ - { name: "images" - data_type: "TYPE_FP32" + data_type: TYPE_FP32 dims: [ 1, 3, 640, 640 ] - } - ] output [ - { name: "output0" - data_type: "TYPE_FP32" + data_type: TYPE_FP32 dims: [ 1, 84, 8400 ] - } - ] - - - instance_group [ { kind: KIND_AUTO