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

fix training in cascade_rcnn (#2386)

上级 588d78b4
......@@ -227,6 +227,8 @@ class CascadeHead(BBoxHead):
clip_box = clip_bbox(boxes_per_image, im_shape[i])
if self.training:
keep = nonempty_bbox(clip_box)
if keep.shape[0] == 0:
keep = paddle.zeros([1], dtype='int32')
clip_box = paddle.gather(clip_box, keep)
rois.append(clip_box)
rois_num = paddle.concat([paddle.shape(r)[0] for r in rois])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册