未验证 提交 150a0d22 编写于 作者: R Ray Liu 提交者: GitHub

Merge pull request #1192 from lijiancheng0614/develop

Update polygon_box_transform op output dims
...@@ -26,7 +26,7 @@ void PolygonBoxTransformCompute(const PolygonBoxTransformParam<CPU>& param) { ...@@ -26,7 +26,7 @@ void PolygonBoxTransformCompute(const PolygonBoxTransformParam<CPU>& param) {
const auto& input_dims = input->dims(); const auto& input_dims = input->dims();
const auto* input_data = input->data<float>(); const auto* input_data = input->data<float>();
auto* output = param.Output(); 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 batch_size = input_dims[0];
int64_t geo_channel = input_dims[1]; int64_t geo_channel = input_dims[1];
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册