未验证 提交 faa4f9a8 编写于 作者: G Guanghua Yu 提交者: GitHub

fix picodet post_process (#5292)

上级 e9cfb187
......@@ -430,9 +430,7 @@ class GFLHead(nn.Layer):
cls_scores, bboxes_reg = gfl_head_outs
bboxes = paddle.concat(bboxes_reg, axis=1)
# rescale: [h_scale, w_scale] -> [w_scale, h_scale, w_scale, h_scale]
im_scale = paddle.concat(
[scale_factor[:, ::-1], scale_factor[:, ::-1]],
axis=-1).unsqueeze(1)
im_scale = scale_factor.flip([1]).tile([1, 2]).unsqueeze(1)
bboxes /= im_scale
mlvl_scores = paddle.concat(cls_scores, axis=1)
mlvl_scores = mlvl_scores.transpose([0, 2, 1])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册