From e609227fc768662e61e451f235c126921c7eafd1 Mon Sep 17 00:00:00 2001
From: lijiancheng0614 <lijiancheng0614@gmail.com>
Date: Wed, 24 Oct 2018 10:30:47 +0800
Subject: [PATCH] update polygon_box_transform op output dims

---
 .../kernel/central-arm-func/polygon_box_transform_arm_func.h    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/operators/kernel/central-arm-func/polygon_box_transform_arm_func.h b/src/operators/kernel/central-arm-func/polygon_box_transform_arm_func.h
index 6db4297046..9cbac1035f 100644
--- a/src/operators/kernel/central-arm-func/polygon_box_transform_arm_func.h
+++ b/src/operators/kernel/central-arm-func/polygon_box_transform_arm_func.h
@@ -26,7 +26,7 @@ void PolygonBoxTransformCompute(const PolygonBoxTransformParam<CPU>& param) {
   const auto& input_dims = input->dims();
   const auto* input_data = input->data<float>();
   auto* output = param.Output();
-  auto* output_data = output->mutable_data<float>();
+  auto* output_data = output->mutable_data<float>(input_dims);
 
   int64_t batch_size = input_dims[0];
   int64_t geo_channel = input_dims[1];
-- 
GitLab