未验证 提交 438889c5 编写于 作者: Z zhanggefan 提交者: GitHub

Update postprocess_cuda.cu (#13263)

fix a bug that affects nms
上级 11104c3d
......@@ -134,7 +134,7 @@ __global__ void filter_kernel(
xmin = fminf(xmin, offset_corners[i * 2 + 0]);
ymin = fminf(ymin, offset_corners[i * 2 + 1]);
xmax = fmaxf(xmin, offset_corners[i * 2 + 0]);
xmax = fmaxf(xmax, offset_corners[i * 2 + 0]);
ymax = fmaxf(ymax, offset_corners[i * 2 + 1]);
}
// box_for_nms(num_box, 4)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册