91fb91fe39f1b510f4027c9409dc4991b64d0823,test_net_cascade.py,,,#,82

Before Change

  if args.mGPUs:
    if args.parallel_type == 0:
      fasterRCNN = nn.DataParallel(fasterRCNN)
    else:
      fasterRCNN.RCNN_base = nn.DataParallel(fasterRCNN.RCNN_base)

  print("load checkpoint %s" % (load_name))

After Change

      args.imdb_name = "coco_2014_train+coco_2014_valminusminival"
      args.imdbval_name = "coco_2014_minival"
      args.set_cfgs = ["ANCHOR_SCALES", "[4, 8, 16, 32]", "ANCHOR_RATIOS", "[0.5,1,2]"]
  elif args.dataset == "imagenet":
      args.imdb_name = "imagenet_train"
      args.imdbval_name = "imagenet_val"
      args.set_cfgs = ["ANCHOR_SCALES", "[8, 16, 32]", "ANCHOR_RATIOS", "[0.5,1,2]"]

Instances