From 1d1dbf76f053bd6f0fccb7e125abd8277efc6f80 Mon Sep 17 00:00:00 2001 From: Bruce Chou Date: Tue, 2 Jun 2020 15:06:12 +0800 Subject: [PATCH] In normal cases channel order is NHWC --- .../tflite/tflite_tensors_to_segmentation_calculator.proto | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mediapipe/calculators/tflite/tflite_tensors_to_segmentation_calculator.proto b/mediapipe/calculators/tflite/tflite_tensors_to_segmentation_calculator.proto index d04aa562b..1ce68e076 100644 --- a/mediapipe/calculators/tflite/tflite_tensors_to_segmentation_calculator.proto +++ b/mediapipe/calculators/tflite/tflite_tensors_to_segmentation_calculator.proto @@ -24,8 +24,8 @@ message TfLiteTensorsToSegmentationCalculatorOptions { } // Dimensions of input segmentation tensor to process. - required int32 tensor_width = 1; - required int32 tensor_height = 2; + required int32 tensor_height = 1; + required int32 tensor_width = 2; required int32 tensor_channels = 3; // How much to use previous mask when computing current one; range [0-1].