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

fix picoheadv2 use_align_head (#5352)

上级 80b1789e
...@@ -466,9 +466,7 @@ class PicoHeadV2(GFLHead): ...@@ -466,9 +466,7 @@ class PicoHeadV2(GFLHead):
), "The size of fpn_feats is not equal to size of fpn_stride" ), "The size of fpn_feats is not equal to size of fpn_stride"
cls_score_list, reg_list, box_list = [], [], [] cls_score_list, reg_list, box_list = [], [], []
for i, fpn_feat, stride, align_cls in zip( for i, (fpn_feat, stride) in enumerate(zip(fpn_feats, self.fpn_stride)):
range(len(self.fpn_stride)), fpn_feats, self.fpn_stride,
self.cls_align):
b, _, h, w = get_static_shape(fpn_feat) b, _, h, w = get_static_shape(fpn_feat)
# task decomposition # task decomposition
conv_cls_feat, se_feat = self.conv_feat(fpn_feat, i) conv_cls_feat, se_feat = self.conv_feat(fpn_feat, i)
...@@ -477,7 +475,7 @@ class PicoHeadV2(GFLHead): ...@@ -477,7 +475,7 @@ class PicoHeadV2(GFLHead):
# cls prediction and alignment # cls prediction and alignment
if self.use_align_head: if self.use_align_head:
cls_prob = F.sigmoid(align_cls(conv_cls_feat)) cls_prob = F.sigmoid(self.cls_align[i](conv_cls_feat))
cls_score = (F.sigmoid(cls_logit) * cls_prob + eps).sqrt() cls_score = (F.sigmoid(cls_logit) * cls_prob + eps).sqrt()
else: else:
cls_score = F.sigmoid(cls_logit) cls_score = F.sigmoid(cls_logit)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册