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

fix mask rcnn target (#5861)

上级 81d28246
......@@ -356,7 +356,7 @@ def generate_mask_target(gt_segms, rois, labels_int32, sampled_gt_inds,
fg_inds_new = fg_inds.reshape([-1]).numpy()
results = []
if len(gt_segms_per_im) > 0:
for j in fg_inds_new:
for j in range(fg_inds_new.shape[0]):
results.append(
rasterize_polygons_within_box(new_segm[j], boxes[j],
resolution))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册