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

fix GFL export model with slice (#4083)

上级 35388eb7
......@@ -419,7 +419,9 @@ class GFLHead(nn.Layer):
mlvl_scores = []
for stride, cls_score, bbox_pred in zip(self.fpn_stride, cls_scores,
bbox_preds):
featmap_size = cls_score.shape[:2]
featmap_size = [
paddle.shape(cls_score)[0], paddle.shape(cls_score)[1]
]
y, x = self.get_single_level_center_point(
featmap_size, stride, cell_offset=cell_offset)
center_points = paddle.stack([x, y], axis=-1)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册