Merge branch 'develop' into add-int8-gemm

f4fbbde8 · Zhen Wang · GitHub · 559241d1 · 96898ba5 · f4fbbde8
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

src/operators/multiclass_nms_op.cpp src/operators/multiclass_nms_op.cpp +2 -1

未找到文件。
--- a/src/operators/multiclass_nms_op.cpp
+++ b/src/operators/multiclass_nms_op.cpp
@@ -32,7 +32,8 @@ void MultiClassNMSOp<Dtype, T>::InferShape() const {
    LOG(kLOG_ERROR) << "Predict bboxes must be equal";
  }
  // pre size, will change in Compute.
-  this->param_.Out()->Resize(framework::make_ddim({input_bboxes_dims[1], 6}));
+  this->param_.Out()->Resize(
+      framework::make_ddim({input_bboxes_dims[1], input_bboxes_dims[2] + 2}));
 }
 }  // namespace operators