From e95cd62c9c48b54fc0663b30aac1a8fd439d77d6 Mon Sep 17 00:00:00 2001 From: wangguanzhong Date: Fri, 9 Dec 2022 16:31:38 +0800 Subject: [PATCH] fix gfl head (#7484) --- ppdet/modeling/heads/gfl_head.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppdet/modeling/heads/gfl_head.py b/ppdet/modeling/heads/gfl_head.py index 66be77089..fdecaf6b0 100644 --- a/ppdet/modeling/heads/gfl_head.py +++ b/ppdet/modeling/heads/gfl_head.py @@ -139,7 +139,7 @@ class DGQP(nn.Layer): axis=2, keepdim=True)], axis=2) else: stat = prob_topk - y = F.relu(self.reg_conv1(stat.reshape([N, 4 * self.reg_topk, H, W]))) + y = F.relu(self.reg_conv1(stat.reshape([N, 4 * self.total_dim, H, W]))) y = F.sigmoid(self.reg_conv2(y)) return y -- GitLab