5bed3f48f0e37d9731b28a6d2505215419d670c8,test_net.py,,,#,77

Before Change

  if args.ngpu > 0:
    cfg.CUDA = True

  fasterRCNN = torch.load(load_name)
  print("load model successfully!")

  if args.ngpu > 0:
    fasterRCNN.cuda()

After Change

          box_deltas = bbox_pred.data
          if cfg.TRAIN.BBOX_NORMALIZE_TARGETS_PRECOMPUTED:
          // Optionally normalize targets by a precomputed mean and stdev
                box_deltas = box_deltas.view(-1, 4) * torch.FloatTensor(cfg.TRAIN.BBOX_NORMALIZE_STDS).cuda() \
                           + torch.FloatTensor(cfg.TRAIN.BBOX_NORMALIZE_MEANS).cuda()
                box_deltas = box_deltas.view(1, -1, 84)
          pred_boxes = bbox_transform_inv(boxes, box_deltas, 1)
          pred_boxes = clip_boxes(pred_boxes, im_info.data, 1)
      else:

Instances