未验证 提交 cc15a9b6 编写于 作者: W wangguanzhong 提交者: GitHub

fix gfl export model (#7462)

上级 e309cf72
...@@ -139,7 +139,7 @@ class DGQP(nn.Layer): ...@@ -139,7 +139,7 @@ class DGQP(nn.Layer):
axis=2, keepdim=True)], axis=2) axis=2, keepdim=True)], axis=2)
else: else:
stat = prob_topk stat = prob_topk
y = F.relu(self.reg_conv1(stat.reshape([N, -1, H, W]))) y = F.relu(self.reg_conv1(stat.reshape([N, 4 * self.reg_topk, H, W])))
y = F.sigmoid(self.reg_conv2(y)) y = F.sigmoid(self.reg_conv2(y))
return y return y
...@@ -435,4 +435,4 @@ class GFLHead(nn.Layer): ...@@ -435,4 +435,4 @@ class GFLHead(nn.Layer):
mlvl_scores = paddle.concat(cls_scores, axis=1) mlvl_scores = paddle.concat(cls_scores, axis=1)
mlvl_scores = mlvl_scores.transpose([0, 2, 1]) mlvl_scores = mlvl_scores.transpose([0, 2, 1])
bbox_pred, bbox_num, _ = self.nms(bboxes, mlvl_scores) bbox_pred, bbox_num, _ = self.nms(bboxes, mlvl_scores)
return bbox_pred, bbox_num return bbox_pred, bbox_num
\ No newline at end of file
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册