提交 7102f227 编写于 作者: L Li Yuanming

fix bug

fix run into exceptions when __C.TEST.BBOX_REG = False
上级 0f970d22
......@@ -321,7 +321,8 @@ if __name__ == '__main__':
pred_boxes = clip_boxes(pred_boxes, im_info.data, 1)
else:
# Simply repeat the boxes, once for each class
pred_boxes = np.tile(boxes, (1, scores.shape[1]))
_ = torch.from_numpy(np.tile(boxes, (1, scores.shape[1])))
pred_boxes = _.cuda() if args.cuda > 0 else _
pred_boxes /= im_scales[0]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册